1. 安装Anaconda最新版
2. 依赖的环境
Anaconda3 x64 (with Python 3.5/3.6)
Windows 64位系统(Windows 7 或 Windows Server 2008 及以上)
CUDA 8 / CUDA 9
cuDNN v5以上
如果安装了CUDA编译的包,请确保你的电脑有Nvidia的显卡。
注:这里没有介绍GPU版本的安装方法,如需要的请搜索其他博文。
安装CUDA 8
(1)首先在官网上(https://developer.nvidia.com/cuda-downloads)下载CUDA,选择自己合适的版本。
该链接界面只显示最新版本。若需要下载以前的版本,可在页面下方点击Legacy Releases,选择自己需要的其他版本,这里安装的是cuda8.0。
这里写图片描述
在Windows中使用迅雷下载,这样比较快。
(2)根据相应的系统选项,我选择的是cuda_8.0.61_win10.exe(大小为1.3G),安装的时候建议选择 自定义 而不是“精简”(从下面的英文解释可以看出,其实这里的精简写成完整应该更贴切,他会安装所有组件并覆盖现有驱动,然而我并不想安装全家桶,何况我的官方显卡驱动比他的新)。
双击安装下载的.exe文件,然后选择解压路径,如下图,解压到哪里无所谓,安装成功会自动删除;
解压完成后,得到如下图:
- 精简:安装所有CUDA模块,并覆盖掉当前的NVIDIA驱动程序;(说实话,容易出问题)
- 自定义:选择自己想要安装的模块,此处选择这里;
选择自定义后,出现下图所示:
下面几个模块准确具体有什么用,不能100%确定,但能大概才出来:
- CUDA:这个是必须的,下面有CUDA Runntime、Samples一些东西;
- NVIDIA GeForce Experience:这个好像是为了更好的游戏体验,之前安装显卡驱动程序时也提示是否安装,果断拒绝了;
- Other components:这里的PhysX好像也是为了游戏体验来的;
- Driver components:这个就要慎重了,意思就是重新安装显卡驱动程序;如果之前已经成功安装驱动程序,这里就不用选了;如果之前没安装驱动程序,建议还是去官网上单独下载驱动程序进行安装吧;
选择好需要安装的模块,这里我都选了,但是第一项cuda中的 Visual Studio 的选项,不用的话,好像就不用安装。
选择安装路径了,我的选择如下图:
- 在如图所示位置建立相应文件夹,然后再指定安装路径;
安装成功后;Ctrl+R,打开cmd:
nvcc -V
输出版本信息,则表明安装成功。
配置环境变量
将F:\CUDA\NVIDIA GPU Computing Tookit\CUDA\v8.0\lib\x64
添加的系统变量的path中;
2. 安装CuDNN
下载对应CUDA 8.0版本的CuDNN:下载链接
(如果安装的是其他版本的CUDA,注意CuDNN的版本)
下载完成后,解压得到一个名为cuda的文件夹;将该文件夹下的文件复制到上一步安装的CUDA中;注意对应的文件夹;
./cuda/bin/**.dll 复制到 ./NVIDIA GPU Computing Tookit/CUDA/v8.0/bin/
./cuda/include/**.dll 复制到 ./NVIDIA GPU Computing Tookit/CUDA/v8.0/include/
./cuda/lib/x64/**.dll 复制到 ./NVIDIA GPU Computing Tookit/CUDA/v8.0/lib/x64/
安装完成;
3. 测试1
使用VS2013打开./cuda/v8.0/Samples_vs2013.sln;
加载完成后,执行本地Windows编译,最后输出成功,见下图;
3. 开始菜单打开Anaconda Prompt,在里面输入conda create -n pytorch python=3.5,为pytorch创建一个虚拟环境
安装cuda参考:
https://blog.csdn.net/weixin_34409703/article/details/93226830
https://blog.csdn.net/yimingsilence/article/details/79126914
https://blog.csdn.net/zhuiqiuzhuoyue583/article/details/88756053
安装Pytorch/torchvision参考:
https://blog.csdn.net/yimingsilence/article/details/79126914
https://blog.csdn.net/Tan915730/article/details/78954482
https://blog.csdn.net/zhuiqiuzhuoyue583/article/details/85327559
https://blog.csdn.net/hpuspring/article/details/79756211
https://zhuanlan.zhihu.com/p/37748231