官网/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

在这里插入图片描述

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐