I am installing PyTorch using conda doing the following:
conda install pytorch torchvision -c pytorch
I have CUDA 9.0 and I have CuDNN 7.0.5 installed into it. I also have an unlinked CUDA 9.2 installation with CuDNN 7.1.4.
Somehow when I install PyTorch 0.4.1, it still picks up the cudnn 7.1 version and donwloads the following package:
pytorch: 0.4.1-py37_cuda9.0.176_cudnn7.1.2_1 pytorch
So, somehow it still pick up on the 7.1 cuddn version and it doesn’t select the 7.0 one. I know for a fact that there is a 7.0 version of PyTorch 0.4.1, since I was able to install it on another PC.
Is there a way to force cuddn version somehow during installation?