I am trying to install PyTorch through anaconda on my machine (runs on Linux, my graphics card is NVIDIA GTX 1080 Ti) using this command:
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
However, even though the installation seems right, the torch.cuda.is_available() command returns False. I checked my driver version and card and they are supposed to be compatible with CUDA 10.2. Here’s what I my nvidia-smi command returns:
±----------------------------------------------------------------------------+
| NVIDIA-SMI 440.36       Driver Version: 440.36       CUDA Version: 10.2     |
±----------------------------------------------------------------------------+
I even tried to downgrade and use earlier version of PyTorch using this command:
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
However, downgrading didn’t work either. After searching stuffs online and trying different methods, I am lost at this point. Any help will be highly appreciated.