先打开cmd控制面板,输入nvcc --version,确认电脑上没有cuda

numba CUDA报错_张量

打开nvidia控制面板,看到显卡型号是3050

numba CUDA报错_张量_02


在帮助–系统信息下打开弹窗,在“组件”下找到CUDA版本,可以看到要下载的是11.4.1版本的

numba CUDA报错_CUDA_03


打开 nvidia的官方cuda下载地址 ,找到11.4.1下载

numba CUDA报错_pytorch_04


选择深绿色的部分,开始下载

numba CUDA报错_pytorch_05


下载后按照默认设置安装即可,其中解压地址可以随便选,安装完之后会把解压的中间文件删除。

重启之后,在cmd中输入命令nvcc --version,发现安装成功

numba CUDA报错_张量_06


接下来吧cuda添加进环境变量里,用windows自带的搜索可以打开

numba CUDA报错_CUDA_07


numba CUDA报错_文件删除_08


在“系统变量”下选择“path”,然后点编辑

numba CUDA报错_CUDA_09


添加以下两条:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\lib\x64

numba CUDA报错_CUDA_10


这样,就可以正常使用CUDA了


报错:RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1

报错:RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1原因:在使用Pytorch加载模型时报错。加载的模型是用两个GPU训练的,而加载模型的电脑只有一个 ...