Hi all,
I am trying to install pytorch 1.4 with torchversion 0.5 that are compatible with CUDA. Every time I install them I get “pytorch 1.40 py3.7_cpu_0 [cpuonly] pytorch” same thing for torchvision.
I have installed cuda 10.1 and it is working with my system. I have uninstalled and install PyTorch multiple time and I only get the cpu only.
I use the following command line “conda install pytorch torchvision cudatoolkit=10.1 -c pytorch”
Please advise and thank you for the help.