Can't run training on PyTorch 1.10.2 with NVidia RTX 3050

#1

I have installed new NVidia RTX 3050, run the training code, and got following message on resnet50 model creation:

/home/tester/anaconda3/lib/python3.8/site-packages/torch/cuda/__init__.py:143: UserWarning: 
NVIDIA GeForce RTX 3050 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_61 sm_70 sm_75 compute_37.
If you want to use the NVIDIA GeForce RTX 3050 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))

PyTorch version 1.10.2, OS: Ubuntu 18.04, Python 3.8.8

Code runs ok on RTX 2060

#2

How was PyTorch installed on this system? It looks like a CUDA 10.2 version was installed; see e.g., Start Locally | PyTorch where CUDA 11.3 should be selected for sm_86.

#3

Yes, CUDA version is 10.2.
I will update and check
Thanks a lot