win10+pytorch 1.1.0 +cuda10.1 + cudnn7.5.0配置
安装 NVIDIA 显卡驱动程序
选择适合自己电脑的显卡驱动下载
安装很简单,直接下一步就可以
安装CUDA 10.1
下载CUDA前,先确认GPU显卡所支持的CUDA版本,
控制面板-》NVIDIA控制面板-》帮助-》系统信息-》组件-》nvidia.dll后面的cuda参数
需要安装VS2013或VS2015或VS2017
安装完成后,命令行输入nvcc -V,如果输出是 CUDA 版本信息,则说明安装成功
安装Cudnn7.5
解压将文件夹中的内容复制到cuda的安装路径下即可。
配置环境变量
控制面板 -> 系统和安全 -> 系统 -> 高级系统设置 -> 高级 -> 环境变量
确认环境变量,CUDA_PATH和CUDA_PATH_V10已经存在
依此添加下列三个路径到Path里
C:\NVIDIA\CUDAv10.1\bin
C:\NVIDIA\CUDAv10.1\include
C:\NVIDIA\CUDAv10.1\lib\x64
注意
一定要更新pip和numpy到最新版本