AssertionError: Torch not compiled with CUDA enabled 解决方案

报错

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

1. 查看本地CUDA支持的版本

nvidia-smi

如红框所示，电脑支持的最高CUDA版本为11.4

2. 检查自己的电脑中CUDA的版本（检查是否安装CUDA）：

   打开cmd，输入

nvcc --version

理论上应显示当前安装的CUDA版本
若显示nvcc不存在，则说明未安装CUDA

1. 进入官网下载CUDA
   建议下载11.3版本
   官网地址：https://developer.nvidia.com/cuda-downloads
   下载历史版本：https://developer.nvidia.com/cuda-toolkit-archive
   找到对应版本进行下载
   选择对应系统
   安装后使用nvcc--version即可看到cuda版本
   
2. 下载 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