报错原因:虚拟环境中的pytorch版本与系统安装的cuda版本不匹配。
解决思路:一种办法是重装指定版本的CUDA,但较高版本的ubuntu无法支持一些低版本CUDA。所以推荐在conda虚拟环境中安装CUDA。
创建虚拟环境并激活:
conda create -n 环境名称
conda activate 环境名称
查看虚拟环境中的Pytorch版本:
import torch
print(torch.__version__)
以下图为例,我们所需安装cuda的版本为11.3:

用conda安装指定 cuda 版本:
conda install cudatoolkit=11.3 -c nvidia
安装cuda-nvcc,注意指定安装版本:
conda install nvidia/label/cuda-11.3.1::cuda-nvcc

注意:若不安装cuda-nvcc会导致调用系统安装的cuda
重新进入环境后再次查看cuda版本,发现cuda版本已经改变,问题解决!
import torch
print(torch.version.cuda)

2599

被折叠的 条评论
为什么被折叠?



