AssertionError: Torch not compiled with CUDA enabled

标题AssertionError: Torch not compiled with CUDA enabled

cuda 和 cudnn 库的卸载与安装

1.先安装最新显卡驱动登录 https://www.nvidia.cn/Download/index.aspx?lang=cn
2.卸载
windows
在windows设置–>应用–>应用和功能中，将带NVDIA并且具有版本号的软件都卸载掉（比如要卸载10.0，那就将下图的软件都卸载）。

进入到环境变量的path变量中，删除旧版本的两个变量。(没有的话就忽略)

删除C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA 下的v10.0文件夹（ cudnn 就在其中）。
3.安装cuda
在 NVIDIA官网 中下载对应版本的 cuda ，打开下载到的安装包cuda_10.1.243_426.00_win10.exe，开始安装。https://developer.nvidia.com/cuda-toolkit-archive

在安装过程中，注意：

选择自定义安装

去掉NVDIA GeForce Experience，因为这东西没用

如果没有安装vs的话，必须去掉在CUDA中的visual studio integration这一项，反正不管怎样去掉就对了。

请确认其他要安装的组件，都满足“新版本”（待安装版本）比“旧版本”（已有版本）要高级的，否请请去掉勾选。

检查
在环境变量中，确认以下两项：

在命令行中输入nvcc -V查看安装的版本。
打开路径 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin 检查nvcc.exe是否存在。
打开路径 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64 检查cupti64_101.dll是否存在。

4. cudnn 库
   下载 cudnn 动态库 （需要先注册），选择 cuda 对应的 cudnn 版本。
   
   将下载到的压缩包解压到 cuda 的安装路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1下。注意解压后的文件夹名为cuda，请将其改成cudnn。
   检查
   在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\cudnn\bin 下确认cudnn64_7.dll动态库。

配置环境变量
在环境变量的path变量中添加：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\cudnn\bin
确认下面4项均存在，如果缺少则添加。
5.安装完cuda后，先卸载torch
pip uninstall torch

6.登录pytorch官网，https://pytorch.org/
找到匹配cuda的pytorch版本进行下载

按下面进行pip,完成