As the title suggests, I have pre-installed CUDA and cudnn (my Tensorflow is using them).
The version of CUDA is 10.0 from
nvcc --version .
The versiuon of cudnn is 7.4 .
I am trying to install pytorch in a conda environment using
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch .
However, the installed pytorch does not detect my GPU successfully.
Does anyone know if there is a way to install GPU-version pytorch with a specific CUDA and cudnn version? I do not want to change CUDA and cudnn version because my Tensorflow is using them.
Any ideas would be appreciated!