Shouldn't Cuda have priority over DML ? #15
Loading…
Reference in New Issue
Block a user
No description provided.
Delete Branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
In utils/device.py
if has_dml():
name = 'dml'
elif torch.cuda.is_available():
name = 'cuda'
No.
has_dml
, rather than dealing with the nightmare of dealing with env vars to nix exposing CUDA capability.