Solved on its own, please ignore
Closed this issue · 2 comments
xiaoboelse commented
PhilippeWeinzaepfel commented
Hi,
I am not familiar at all with cuda/pytorch/compilation on windows.
However, it seems that your installed version of pytorch is a cpu-only build. If you have installed it with conda, do not forget the nvidia channel and the pytorch-cuda, eg conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
.
Then you need to install the cuda toolkit (https://developer.nvidia.com/cuda-downloads) and set CUDA_HOME to the location where it is installed.
Best regards,
Philippe
xiaoboelse commented
ok, Thank you very much for the friendly reminder. It has been installed