Quantcast
Channel: Active questions tagged python - Stack Overflow
Viewing all articles
Browse latest Browse all 23160

Conda Install PyTorch Results in Incorrect Version

$
0
0

I'm trying to install PyTorch through Conda. However, when I run the following command according to the PyTorch documentation:

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

it does not seem to install the correct pytorch-cuda version. That is, running

>>> torch.version.cuda'12.1'

This is a problem because then when I run

>>> torch.cuda.is_available()UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 11020). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver. (Triggered internally at ../c10/cuda/CUDAFunctions.cpp:108.)  return torch._C._cuda_getDeviceCount() > 0

How do I fix this?

I tried rearranging the order of the packages. I also tried changing the version of Python from 3.8.19 -> 3.8.18.


Viewing all articles
Browse latest Browse all 23160

Trending Articles



<script src="https://jsc.adskeeper.com/r/s/rssing.com.1596347.js" async> </script>