Post on: May 9, 2024Last edited: Jul 4, 2024Words 00 min

type
status
date
slug
summary
tags
category
icon
password
😀
NVIDIA GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation. 在中国云服务器使用torch时,可能遇到的torch版本与CUDA不兼容的问题

问题分析

根据报错信息的描述,我们可以很明显地发现是由于torch版本与CUDA版本不匹配导致的。
如果你是在中国的云服务器,例如featurize、AutoDL等,即使你按照python程序的依赖安装了对应版本torch,也可能会报这个错误,这是因为中国的云服务器大多数默认是清华镜像源下载,这样可以有更高的下载速率。然而清华的镜像源提供的torch只能支持cpu,所以这里的错误是未安装GPU版本的torch。
此时需要到torch官网,按照你的需要找到对应的安装指令,这样的指令后面会跟着它指定的下载地址。
例如,对CUDA11.1的torch1.9.0
请注意,在安装之前应当卸载你已经安装的torch、torchvision、torchaudio

  • Twikoo
在Linux中重新安装CUDA和cnDNN

🌸在Linux中重新安装CUDA和cnDNN

在Linux中重新安装CUDA和cnDNN,同样适用于类似Featurize云平台更换到指定的CUDA版本,例如CUDA11.1


Announcement
🎉欢迎来到我的博客🎉
👏欢迎更新体验👏