Pytorch on GeForce RTX 3090 GPUs

Hi, I am working with GeForce RTX 3090 GPUs. Recently, I am facing an error that I have never seen before. When creating a new virtual env, upgrading pip, and trying to install PyTorch by

pip3 install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

I am getting this error:

ERROR: Could not find a version that satisfies the requirement torch==1.11.0+cu113 (from versions: 1.0.0, 1.0.1, 1.0.1.post2, 1.1.0, 1.2.0, 1.3.0, 1.3.1, 1.4.0, 1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.0+cu113, 1.10.1, 1.10.1+cu113, 1.10.2, 1.10.2+cu113)
ERROR: No matching distribution found for torch==1.11.0+cu113

I can install nightly by pip3 install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu113/torch_nightly.html, but then I face this error:

/cis/home/shraman/multimodal_self_supervision/experiment_codes/ssl_env/lib/python3.6/site-packages/torch/cuda/__init__.py:143: UserWarning:
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 GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

and the code stops
RuntimeError: CUDA error: no kernel image is available for execution on the device

I have used the same GPUs many many times with the different virtual environments a month ago. What is the problem now?

I think PyTorch 1.11 is 3.7+ only, as Python 3.6 is not supported by Python.org anymore.

Best regards

Thomas

1 Like