Hello, I’m getting following error:
NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
If you want to use the NVIDIA GeForce RTX 3090 GPU with PyTorch, please check the instructions at Start Locally | PyTorch 1
I’ve tried to install compatible version but I don’t think I’m doing it correctly. Can anyone help me to get it installed correctly?
$ python3 -m pip list | grep torch
torch                   1.10.1              
torchvision             0.11.2

$ which pip3
/usr/bin/pip3

CUDA Version                              : 11.4