NVIDIA GeForce RTX 3080 Ti with CUDA capability sm_86 is not compatible with the current PyTorch installation

Nam_Truong_Tran: <p>Hey PyTorch Community,</p>
<p>currently i working on a project to train an agent.<br>
Unfortunately i can’ run my script due to this error:</p>
<p>NVIDIA GeForce RTX 3080 Ti with CUDA capability sm_86 is not compatible with the current PyTorch installation.<br>
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.<br>
If you want to use the NVIDIA GeForce RTX 3080 Ti GPU with PyTorch, please check the instructions at <a href="https://pytorch.org/get-started/locally/" class="inline-onebox" rel="noopener nofollow ugc">Start Locally | PyTorch</a></p>
<p>I using Ubuntu 20.04.</p>
<p>Informations:</p>
<pre><code class="lang-auto">RUN nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243
</code></pre>
<pre><code class="lang-auto">RUN python -c "import torch; print(torch.__version__)"
1.9.0+cu111
</code></pre>
<pre><code class="lang-auto">RUN nvidia-smi
Mon May 15 20:02:00 2023       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.182.03   Driver Version: 470.182.03   CUDA Version: 11.4     |
|-------------------------------+----------------------+----------------------+
| 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 ...  Off  | 00000000:2D:00.0  On |                  N/A |
|  0%   38C    P8    26W / 350W |    602MiB / 12045MiB |      8%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1125      G   /usr/lib/xorg/Xorg                175MiB |
|    0   N/A  N/A      1416      G   /usr/bin/gnome-shell               33MiB |
|    0   N/A  N/A     22053      G   ...464272974724260903,262144      124MiB |
|    0   N/A  N/A     23840      G   .../ros/noetic/lib/rviz/rviz       16MiB |
|    0   N/A  N/A     24776      C   python                            247MiB |
+-----------------------------------------------------------------------------+
</code></pre>
<p>Thank you very much ! <img src="https://discuss.pytorch.org/images/emoji/apple/slight_smile.png?v=12" title=":slight_smile:" class="emoji" alt=":slight_smile:" loading="lazy" width="20" height="20"></p>

ptrblck: <p>Could you update PyTorch to the latest stable or nightly release as it’s supporting your GPU.<br>
Also, based on the error message it seems you are using a PyTorch binary with CUDA 10.2 which doesn’t match your debug print output.</p>