报错
Traceback (most recent call last):
File "main_pretrain.py", line 250, in <module>
main(args)
File "main_pretrain.py", line 185, in main
model.to(device)
File "D:\APP\Anaconda3\envs\py37\lib\site-packages\torch\nn\modules\module.py", line 899, in to
return self._apply(convert)
File "D:\APP\Anaconda3\envs\py37\lib\site-packages\torch\nn\modules\module.py", line 570, in _apply
module._apply(fn)
File "D:\APP\Anaconda3\envs\py37\lib\site-packages\torch\nn\modules\module.py", line 570, in _apply
module._apply(fn)
File "D:\APP\Anaconda3\envs\py37\lib\site-packages\torch\nn\modules\module.py", line 593, in _apply
param_applied = fn(param)
File "D:\APP\Anaconda3\envs\py37\lib\site-packages\torch\nn\modules\module.py", line 897, in convert
return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking)
File "D:\APP\Anaconda3\envs\py37\lib\site-packages\torch\cuda\__init__.py", line 208, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled
解决
安装cuda
- 查看本地CUDA支持的版本
nvidia-smi
如红框所示,电脑支持的最高CUDA版本为11.4
-
检查自己的电脑中CUDA的版本(检查是否安装CUDA):
打开cmd,输入
nvcc --version
理论上应显示当前安装的CUDA版本
若显示nvcc不存在,则说明未安装CUDA
- 进入官网下载CUDA
建议下载11.3版本
官网地址:https://developer.nvidia.com/cuda-downloads
下载历史版本:https://developer.nvidia.com/cuda-toolkit-archive
找到对应版本进行下载
选择对应系统
安装后使用nvcc--version
即可看到cuda版本
- 下载 cuDNN
地址:https://developer.nvidia.com/rdp/cudnn-download
下载cuDNN是需要登录英伟达开发者账户
注意:必须选择和安装的CUDA匹配的版本
解压
复制bin下的文件 cudnn64_8.dll 到
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin
添加环境变量
运行test