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
- Author:Kecilimu
- URL:https://kecilimu-notion.vercel.app//article/20240509133233
- Copyright:All articles in this blog, except for special statements, adopt BY-NC-SA agreement. Please indicate the source!