整理gpu加速的深度学习框架pytorch的安装过程。

0,资源整理

1,百度网盘完整版资源(包括:visual strudio+显卡驱动+cuda+cudnn+pytorch==0.4.1的.whl安装包,我安装的就是这一套,现将这些资源整理出来)

链接:https://pan.baidu.com/s/1Wuyhm6r8sh9-KloeMq4dDA

提取码:akj7

2,各资源来源:

visual studio(2017社区版就可以,2015/2013没有试过)

pytorch安装包(注意对应的cuda版本和对应的操作系统)

显卡驱动下载

CUDA(这个要与pytorch安装包相配)

CUDNN(这个要与CUDA相配)

3,参考博客:

参考1:解决pytorch官网下载慢ubuntu16.04+anaconda3(python3.6)+pytorch0.4.1+cuda9.0+cudnn7.1安装指南(关键要知道0.4.1对应的cuda和cudnn版本)

参考2:【PyTorch】windows10 + CUDA 9.1 + cuDNN v7.0.5 + PyTorch 0.4.1(关键要知道0.4.1对应的cuda和cudnn版本)

参考3:新电脑重新安装win10+python3.6+anaconda+tensorflow1.12(gpu版)(关键要学习cuda、cudnn的安装及环境变量的配置)

参考4:Windows下安装PyTorch0.4.0(几个在线pip下载.whl安装命令,但是不用VPN基本都卡死了)

4,可以直接用百度网盘的压缩包,所有安装包都是齐全的;也可以自己从各个网站自己下载

1,安装最新版的显卡驱动

1,将所有带NVIDIA的驱动全删了



2,使用与自己显卡匹配的驱动安装包,安装最新版的驱动

2,安装visual studio

使用community版本就可以,我这里用的2017社区版成功了,安装过程中不要改路径,不要改配置,全部默认配置安装下来就行。安装过程要联网。

3,安装cuda

选择自己需要的版本->选择自己的操作系统->选择exe(local)安装->将Base Installer和所有Patch都下载下来->按顺序全部安装一遍,安装过程不要改路径,全部选择精简安装(其实也就是默认安装)

如果从百度网盘下载的,直接按顺序安装一遍就行了



4,安装cudnn,配置环境变量

1,选择对应cuda版本同时满足自己所需要版本的cudnn,下载下来,解压缩(第一次下载要加入一个什么计划,申请个账号就行了)



2,将解压缩后的三个文件夹复制到安装CUDA9.0的路径覆盖原有的文件。本机的安装路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0,直接复制粘贴过来就行。



3,配置环境变量。桌面时右键“此电脑”,选择“属性”打开系统窗口,然后选择并打开“高级系统设置”进入系统属性窗口,打开环境变量,然后将

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin,

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include,

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64,

这三个路径填到系统变量的Path中,这样环境变量就配置好了。

5,安装pytorch

1,下载安装包。因为我是使用pip安装,并且cuda为9.0,因此点击对应链接



根据对应python版本和操作系统下载安装包,比如我要下载torch0.4.1,环境为python3.6,windows64位,就下载torch-0.4.1-cp36-cp36m-win_amd64.whl。下载速度快的时候很快,慢的时候很慢,也不知道为啥。



2,pip离线安装

cmd输入命令:pip install ...(安装包保存路径)\torch-0.4.1-cp36-cp36m-win_amd64.whl

6,安装torchvision

直接pip在线安装,可以配置一下镜像加速。pytorch0.4.1对应torchvision0.2.1

pip install torchvision==0.2.1

7,验证

最新文章

  1. 2016年11月28日--ADO.Net 查、插、删、改 小练习
  2. html 上传预览图片
  3. Javascript中的正则表达式
  4. 第8章 用户模式下的线程同步(2)_临界区(CRITICAL_SECTION)
  5. .gitignore 配置
  6. 将Tomcat注册为Windows服务
  7. IE9-10 option BUG
  8. MVC运行原理
  9. linux 以root用户登录ftp
  10. [多线程同步练习]PV操作
  11. 微软新一代输入法框架 TSF - Text Service Framework 小小的研究
  12. 总结scala(一)
  13. 【LeetCode】118. Pascal's Triangle
  14. UWP中使用Composition API实现吸顶(1)
  15. (翻译).NET应用架构
  16. 暴雪《争霸艾泽拉斯》*采用自适应 SSAO
  17. db2报错信息
  18. 这一周~&&html+css的学习感悟
  19. 【Java】Springboot-Quartz-分布式任务调度
  20. JavaScript资源收集分享,持续更新中。。。

热门文章

  1. javascript module system all in one
  2. JavaScript interview Question - Create a Array with two papameters without using loop!
  3. queueMicrotask & microtask
  4. bind & this & new & arrow function
  5. React & react-native & vue & cli & environment information & report bugs
  6. Learn-JavaScript-with-MDN 系列文章: 01. var & let & const 对比
  7. SSL/TLS协议详解(下)——TLS握手协议
  8. 详解Go语言调度循环源码实现
  9. selenium之元素定位的方法(一)
  10. Linux:使用systemd管理进程