I have a gpu on my pc, but it does not show up when I use,
print([torch.cuda.device(i) for i in range(torch.cuda.device_count())])
is there a further setting required to use the gpu
these are some further details regarding the gpu on my pc
Chipset Model: Apple M1
Total Number of Cores: 8
The GPU on Mac is not Nvidia’s kind. It is M1 GPU designed by Apple.
You need to first set the device to
See this for more details on installation and usage:
Running PyTorch on the M1 GPUtype or paste code here
thanks for the info, it now detects the gpu,
conda install pytorch -c pytorch-nightly
mps_device = torch.device("mps")
x = torch.ones(1, device=mps_device)
print ("MPS device not found.")
UserWarning: The operator 'aten::nonzero' is not currently supported on the MPS backend and will fall back to run on the CPU. This may have performance implications.
This is a
UserWarning. You can ignore it for not. Most likely, some bug in the current
nightly version. Don’t worry about it.
I have the same issue with current stable version. Moreover, it seems that all computations are going to be on CPU.