Detected that PyTorch and torchvision were compiled with different CUDA versions

Hi,

There are many version issue in cuda and pytorch.

RuntimeError: Detected that PyTorch and torchvision were compiled with different CUDA versions. PyTorch has CUDA Version=11.7 and torchvision has CUDA Version=11.8. Please reinstall the torchvision that matches your PyTorch install.

How can I find a proper way to install them?
from Start Locally | PyTorch
CUDA only shows 11.7 (not 11.8). I need 11.7 I guess
But this command cannot fix my problem.
Is there any solution? Thx

pip3 install torch torchvision torchaudio
nvidia-smi
Tue Sep 26 02:46:31 2023       
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.104.05             Driver Version: 535.104.05   CUDA Version: 12.2     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|
|   0  NVIDIA GeForce RTX 4090        On  | 00000000:01:00.0  On |                  Off |
|  0%   45C    P8              26W / 450W |    361MiB / 24564MiB |      3%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+
                                                                                         
+---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
|    0   N/A  N/A      1312      G   /usr/lib/xorg/Xorg                          126MiB |
|    0   N/A  N/A      1593      G   /usr/bin/gnome-shell                         41MiB |
|    0   N/A  N/A      3812      G   ...irefox/3131/usr/lib/firefox/firefox      173MiB |
+---------------------------------------------------------------------------------------+
pip3 install torch torchvision torchaudio
Requirement already satisfied: torch in /home/ak/anaconda3/lib/python3.9/site-packages (2.0.1+cu118)
Requirement already satisfied: torchvision in /home/ak/anaconda3/lib/python3.9/site-packages (0.15.2+cu118)
Requirement already satisfied: torchaudio in /home/ak/anaconda3/lib/python3.9/site-packages (2.0.2+cu118)
Requirement already satisfied: filelock in /home/ak/anaconda3/lib/python3.9/site-packages (from torch) (3.9.0)
Requirement already satisfied: typing-extensions in /home/ak/anaconda3/lib/python3.9/site-packages (from torch) (4.7.0)
Requirement already satisfied: sympy in /home/ak/anaconda3/lib/python3.9/site-packages (from torch) (1.11.1)
Requirement already satisfied: networkx in /home/ak/anaconda3/lib/python3.9/site-packages (from torch) (3.1)
Requirement already satisfied: jinja2 in /home/ak/anaconda3/lib/python3.9/site-packages (from torch) (2.11.3)
Requirement already satisfied: triton==2.0.0 in /home/ak/anaconda3/lib/python3.9/site-packages (from torch) (2.0.0)
Requirement already satisfied: cmake in /home/ak/anaconda3/lib/python3.9/site-packages (from triton==2.0.0->torch) (3.25.0)
Requirement already satisfied: lit in /home/ak/anaconda3/lib/python3.9/site-packages (from triton==2.0.0->torch) (15.0.7)
Requirement already satisfied: numpy in /home/ak/anaconda3/lib/python3.9/site-packages (from torchvision) (1.23.5)
Requirement already satisfied: requests in /home/ak/anaconda3/lib/python3.9/site-packages (from torchvision) (2.28.2)
Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in /home/ak/anaconda3/lib/python3.9/site-packages (from torchvision) (9.4.0)
Requirement already satisfied: MarkupSafe>=0.23 in /home/ak/anaconda3/lib/python3.9/site-packages (from jinja2->torch) (2.0.1)
Requirement already satisfied: charset-normalizer<4,>=2 in /home/ak/anaconda3/lib/python3.9/site-packages (from requests->torchvision) (2.0.4)
Requirement already satisfied: idna<4,>=2.5 in /home/ak/anaconda3/lib/python3.9/site-packages (from requests->torchvision) (3.4)
Requirement already satisfied: urllib3<1.27,>=1.21.1 in /home/ak/anaconda3/lib/python3.9/site-packages (from requests->torchvision) (1.26.16)
Requirement already satisfied: certifi>=2017.4.17 in /home/ak/anaconda3/lib/python3.9/site-packages (from requests->torchvision) (2023.7.22)
Requirement already satisfied: mpmath>=0.19 in /home/ak/anaconda3/lib/python3.9/site-packages (from sympy->torch) (1.3.0)
DEPRECATION: torchsde 0.2.5 has a non-standard dependency specifier numpy>=1.19.*; python_version >= "3.7". pip 23.3 will enforce this behaviour change. A possible replacement is to upgrade to a newer version of torchsde or contact the author to suggest that they release a version with a conforming dependency specifiers. Discussion can be found at https://github.com/pypa/pip/issues/12063

This one solve my problem.

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
1 Like

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.