CODEWITHSUNDEEP
installationpytorchconda
Wei
Wei

Reputation: 315

Install Pytorch GPU with pre-installed CUDA and cudnn

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!

Upvotes: 2

Views: 2683

Answers (1)

Wei
Wei

Reputation: 315

So I solved this myself finally. The issue is that I didn't reboot my system after installing pytorch. After rebooting, torch.cuda.is_available() returns True as expected.

Upvotes: 1

Related Questions