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

numba CUDA报错_张量

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

numba CUDA报错_pytorch_02


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

numba CUDA报错_CUDA_03


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

numba CUDA报错_pytorch_04


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

numba CUDA报错_深度学习_05


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

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

numba CUDA报错_pytorch_06


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

numba CUDA报错_pytorch_07


numba CUDA报错_CUDA_08


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

numba CUDA报错_pytorch_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报错_文件删除_10


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