步骤:
1、查看显卡驱动版本:NVIDA控制面板—系统信息—显示—驱动程序版本,在(https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html)上查看对应 cuda 版本。
2、安装 cuda:在(https://developer.nvidia.com/cuda-92-download-archive)上下载,默认安装路径安装即可。
3、添加环境变量:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\lib\x64
4、安装 cudnn:在(https://developer.nvidia.com/rdp/cudnn-archive)下载对应版本cudnn, 需要先注册一个账号,下载后:
cuda\bin目录中的 cudnn64_7.dll 复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\bin 中
将cuda\include目录中的 cudnn.h 复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\include 中
将cuda\lib\x64 目录中的 cudnn.lib 复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\lib\x64 中
5、conda创建一个虚拟环境
6、安装 pytorch:运行conda install pytorch cudatoolkit=9.2 -c pytorch即可。
此方法指定CUDA Toolkit版本后,conda会自动匹配到合适版本的PyTorch 。
Win10+anaconda+cuda9.2+cudnn+pytorch
最新推荐文章于 2022-03-18 13:09:19 发布