win10下安装使用pytorch以及cuda9、cudnn7.0

pytorch是一个优雅的深度学习库,相比TensorFlow更年轻也更充满潜力,在官方的介绍中pytorch只支持linux和mac,但其实windows也是可以安装以及正常使用的

平台: win10(版本1709) CPU:i5-7400 显卡:1060 6G 内容:8G

软件: anaconda3 pycharm专业版

首先从官网下载相应的cuda和cudnn库,记住下载win10版本的,不要下错。一些额外的下载信息可以查看我之前的一篇文章(pytorch-0.2成功调用GPU:ubuntu16.04,Nvidia驱动安装以及最新cuda9.0与cudnnV7.0配置):http://blog.csdn.net/iamoldpan/article/details/78459208

windows下安装cuda和cudnn还是非常容易的,下载最新版即可(cuda9.0和与之相对应的cudnn版本)

安装好之后,开始创建虚拟环境,这里建议使用anaconda3,不论是方便性还是拓展性都比只使用pip强。在官网下载anaconda3然后进行安装,创建一个虚拟环境:http://blog.csdn.net/iamoldpan/article/details/77969477(注意在安装的时候需要勾上环境变量的选项,如果没有需要自行在环境变量中加入相应路径(…/anaconda3/Scripts))

创建好虚拟环境后,在pycharm中创建工程,将解释器选择为之前创建的环境(不用pycharm也可以,这里只是做例子,只要有python的简单IDE即可):

上图是已经安装完成后的解释器界面,可以看到pytorch和torchvision以及一系列依赖包都已经安装好。

选择解释器后,开始conda环境中只有刚开安装的几个程序包,这是需要安装pytorch需要的依赖包:

注意:一定要在虚拟环境中进行安装,如果在终端中创建了虚拟环境但是无法激活,可以通过conda install -n my-env package指令虚拟环境来进行安装,同样pip安装直接使用虚拟环境中的pip命令进行安装:.../envs/my-envs/Scripts/pip install package,总之要安装到虚拟环境中去。

conda install numpy pyyaml mkl setuptools cmake cffi

因为直接从官网源下载速度很慢,我们可以修改conda的安装源来进行加速:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

然后进行依赖包的安装

安装完依赖包后,开始进行pytorch和torchvision的安装: 首先下载pytorch的pip安装包:https://pan.baidu.com/s/1nvaamrn#list/path=%2F 我安装的是,pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2,即python3.6-cuda8版本的安装包下好后,进入到该文件的目录中,在终端中输入:

pip install pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2

即可进行安装,安装速度一般很快,如果是固态硬盘,5~6s即可安装完毕。 torchvision的安装相对简单很多,是从conda和pip安装即可,只要注意安装到对应的虚拟环境中即可。

大功告成

全部安装完后,如果没有意外:

就可以在window中使用pytorch了。

此文由腾讯云爬虫爬取,文章来源于Oldpan博客

欢迎关注Oldpan博客公众号,持续酝酿深度学习质量文:

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

编辑于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏haifeiWu与他朋友们的专栏

美团外卖开源路由框架 WMRouter 源码分析

上周四美团外卖技术团队开源了一个 Android Router 的框架: WMRouter,博客详细介绍了用法以及设计方案,还不熟悉的同学可以先去看一下。本篇博...

47310
来自专栏FSociety

Python实时爬取斗鱼弹幕

斗鱼提供的文档已经是一年前的了,里面传回的消息内容增加了不少,但整体逻辑还是没变,我这边只取了弹幕里面的昵称和文本内容,其他的消息各位可以先打印出来看了再写正则...

65810
来自专栏前端知识分享

第206天:http协议终极详解---看这一篇就够了

HTTP协议是Hyper Text Transfer Protocol(超文本传输协议)的缩写,是用于从万维网(WWW:World Wide Web )服务器传...

82940
来自专栏前端知识分享

第119天:移动端:CSS像素、屏幕像素和视口的关系

移动前端中常说的 viewport (视口)就是浏览器显示页面内容的屏幕区域。其中涉及几个重要概念是 dip ( device-independent pixe...

20950
来自专栏曾大稳的博客

HashMap源码分析

HashMap是一个很经典的键值对集合,从它的广泛应用程度和源码的学习角度上我们不得不去解析它。 我们先看一下HashMap的存储结构((图片均来源于网络)),...

11420
来自专栏前端知识分享

第145天:jQuery.touchSlider触屏满屏左右滚动幻灯片

30620
来自专栏haifeiWu与他朋友们的专栏

Android动态权限申请(Kotlin版)

Android6.0之后,APP有些用到的权限需要动态申请,虽然这个功能已经出来好久了,但一直没有研究过,只知道用法简单,到时候copy一下就好了,github...

1.3K20
来自专栏曾大稳的博客

Glide v3.7源码分析(3)------GenericRequestBuilder.into(时序图)

16730
来自专栏前端知识分享

第210天:node、nvm、npm和gulp的安装和使用详解

它不是JS文件,也不是JS框架,而是Server side JavaScript runtime,当服务端的一个JS文件运行时,会被NODE拦截,在NODE中运...

27710
来自专栏零基础使用Django2.0.1打造在线教育网站

零基础使用Django2.0.1打造在线教育网站(二十):课程相关页面配置

努力与运动兼备~~~有任何问题可以加我好友或者关注微信公众号,欢迎交流,我们一起进步!

35610

扫码关注云+社区

领取腾讯云代金券

年度创作总结 领取年终奖励