您的位置:首页 > 编程语言 > Python开发

解决pytorch官网下载慢ubuntu16.04+anaconda3(python3.6)+pytorch0.4.1+cuda9.0+cudnn7.1安装指南

2018-12-07 16:54 1901 查看

一.准备工作

1.系统环境是ubuntu

2.去anaconda官网下载anaconda3,python版本是3.5以上的就行,官网最新的是python3.7,没有关系,反正自己可以创建新环境

来选择其他版本的python.

3.去清华镜像源下载对应版本的pytorch,这里作者使用的是pytorch-0.4.1-py36_cuda9.0.176_cudnn7.1.2_1.tar.bz2

这个文件名意思是:python3.6版本,cuda9.0,cudnn7.1.(旧版本的pytorch的函数名与新版本的不匹配,尽量选择0.3以上

版本pytorch)

二.安装

1.首先安装anaconda,具体安装方式网上查找,添加环境变量,这样才能在控制端输入"conda"指令.

2.创建环境"mypytorch":

[code]conda create -n mypytorch python=3.6

3.进入环境

[code]source activate mypytorch

4.然后在mypytorch环境中利用conda命令安装下载好的pytorch文件

[code]conda install /home/user/pytorch-0.4.1-py36_cuda9.0.176_cudnn7.1.2_1.tar.bz2

5.安装完pytorch后,在继续conda 安装pytorch,为什么安装两遍呢,因为如果直接利用conda安装,会从官网下载安装,速度太慢

(你翻墙也行,速度快可以忽略这个方法),第二次是利用pytorch官网给的命令,当conda发现本地已经装了pytorch后,他不会再

安装了,但是会把pytorch需要的其他依赖包给装好,比如cuda9.0等,这些包是从系统设置的源中下载的(比如aliyun)

[code]conda install pytorch torchvision -c pytorch

6.这样pytorch就安装完了

 

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐