Window10+Anaconda(python3.7)pytorch1.0.1+cuda9.0.176+cudnn-v7.5.0

前情提要

拖了好久的深度学习终于提上了日程,听说pytorch比较好用,而且正好实验室里有相关的书,就不管三七二十一看了起来。     第一关当然是配置环境咯,python基本环境当然是必须的,因为我的电脑上有ArcGIS自带的python2.7、Anaconda3的python3.7、还有一个不知道什么时候装的python3.6,所以我一度担心这些小家伙们会在一起打架,后来试探性地在命令行中进入不同版本的python发现完全无影响。

于是,接下来就愉(tan)快(te)地安装起来啦。

安装流程

1. 你先有anaconda(当然没有也行,直接用pip安装的话那就百度吧,我没用过),这个时候你如果没有欲望使用GPU来加速你的运算,或者你的老板给你们购置的电脑没有英伟达爸爸的显卡,那么直接打开你命令行,并在pytorch的官网上按照下图选择,接下来等待就好,很愉快。

2.安装GPU版本的pytorch就要相对复杂了:

首先,确认一下你的NVIDIA显卡是否支持CUDA;

第二,到英伟达官网下载cuda,链接给的是我安装的cuda9.0 -->https://developer.nvidia.com/cuda-90-download-archive

选择你的设备对应的信息,需要注意的是:Installer Type选择[local]版本。cuda的安装十分简单,下载下安装包运行,默认路径就好,如果你不想让它在你的C盘,那请记住你的安装路径,后面配置cuDNN时还会用到。

第三,就是下载cuDNN,这是英伟达官网上的cuDNN下载地址-->https://developer.nvidia.com/cudnn不过需要你登录,如果你有NVIDIA账号直接登录就好,没有的话邮箱注册也很快。登陆上以后会出现以下界面,勾选他们的balabala你就看到下载选项了。选择forCUDA9.0-->for Windows10.

下载下来一会是一个压缩包解压以后,得到下图的文件,回头去找到你的CUDA安装路径里也有这对应的三个文件夹,把里面的文件拷贝到CUDA安装路径下,(注意!不要直接拷贝这三个文件夹去替换安装路径下的文件夹啊)

拷贝完成以后基本上cuda这一部分就安装完成啦!

第四,打开你的命令行,运行 conda install pytorch torchvision cudatoolkit=9.0 -c pytorch,这行命令就是下载window10+anaconda(python3.7)+cuda9.0+cuDNN7.5下的pytorch,(如果下载中断,多运行几遍上面的conda语句,总会成功的[微笑],或者试着开开蓝灯之类的fq工具)。等待下载完成以后,测试你的torch是否成功下载到你的电脑,并且同时测试你的cuda是否能用,在python下运行以下命令:

如果你的结果也是上图这样的话,恭喜你,请在pytorch的海洋中遨游吧!

文档写的仓促,也是在借鉴了别人的基础上,自己踩完了坑出来的,如果你也有什么疑问或者同样在学习pytorch欢迎留言讨论!

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值