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 条评论
登录 后参与评论

相关文章

来自专栏乐沙弥的世界

Percona XtraDB Cluster(PXC)掉电无法正常启动

办公室掉电,PXC集群环境无法启动,也就是说整个集群的状态处于丢失的情形。因此需要采取强制的方式来进行,见下面的描述。

602
来自专栏散尽浮华

Gitblit版本服务器环境部署记录

Gitblit介绍 Gitblit 是一个纯 Java 库用来管理、查看和处理 Git 资料库.相当于 Git 的 Java 管理工具,支持linux系统。 G...

2685
来自专栏程序猿DD

spring-boot-starter-swagger迎新伙伴支持,加速更新进度(1.3.0.RELEASE)

从该starter创建至今收到了不少使用反馈,同时也有不错的童鞋申请加入一起维护。本篇先欢迎小火童鞋的加入及贡献,接下来具体说说本次的更新内容。 本次更新主要新...

3315
来自专栏nnngu

013 BIO、NIO、AIO的区别

IO通常分为几种,BIO(阻塞 Blocking IO)、NIO(非阻塞 Non-Blocking IO)、AIO(异步非阻塞)。 BIO 在JDK1.4出来之...

5069
来自专栏三杯水

排查logstash2.4升级到5.0版本后kafka不兼容问题

/usr/share/logstash/vendor/bundle/jruby/1.9/gems/logstash-input-kafka-5.0.5/CHAN...

781
来自专栏灯塔大数据

15 个 Android 通用流行框架大全

? 1 缓存 名称描述DiskLruCacheJava实现基于LRU的磁盘缓存 2 图片加载 名称描述Android Universal Image Lo...

2846
来自专栏java一日一条

15 个 Android 通用流行框架大全

Android自带很多测试工具:JUnit,Monkeyrunner,UiAutomator,Espresso等

792
来自专栏AI-vell

音频打标签工具

代码在:https://github.com/vell001/audio-annotator

32210
来自专栏zhisheng

Java研发方向如何准备BAT技术面试答案(中)

JavaEE: 1.servlet生命周期及各个方法 参考文章 http://www.cnblogs.com/xuekyo/archive/2013/02/2...

39313
来自专栏PHP在线

开始使用PHPUnit

想试一下单元测试很久了,但是一直没有合适的项目,而且中文网络上的PHPUnit教程要么太旧,要么太乱,所以总也没学会。最近准备把手头的项目重构一下,决定开始使用...

1243

扫码关注云+社区