官网/pytorch-cuda

pytorch官网

官网/cuda-gpu

nvidia官网

问题/sm_86

显卡支持的cuda算力

问题/NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.

问题：

/home/ouc/all_users/ransj/miniconda3/lib/python3.8/site-packages/torch/cuda/__init__.py:146: UserWarning: 
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 https://pytorch.org/get-started/locally/

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

原因

当前NVIDIA GeForce RTX 3090是CUDA sm_86的能力，并没有被当前版本的pytorch-cuda支持

解决：

更换支持更高版本的CUDA的pytorch
卸载原来的torch
	pip uninstall torch torchvision torchaudio
进入官网
	https://pytorch.org/get-started/locally/
安装支持更高版本的cuda的pytorch