ubuntu中pytorch安装速度太慢的解决方法

ubuntu中pytorch安装速度太慢的解决方法

一，电脑配置

1. ubuntu_18.04
2. cuda9.0
3. gpu
4. conda

二，切换到国内镜像
解释一下，pytorch的官网中有安装教程，链接，但是按着下载老是会中断。
1.切换到国内镜像，终端输入

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

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --set show_channel_urls yes

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

三，创建pytorch环境
1.创建环境

conda create -n pytorch python=3.6     #可以自行修改为别的版本

2.激活环境

source activate pytorch

退出环境也附带写在这里，但现在还是要在pytorch环境中

source deactivate pytorch

3.安装pytorch
首先我们进入pytorch的官网，选择合适自己的版本，比如我是基于conda安装，cuda版本是9.0。
只需要选择对应指令安装即可，比如我的是：

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

但是在安装过程中发现这样安装速度依旧很慢，pytorch的包依旧没法下载，原因是在上面的指令中 -c pytorch是指从pytorch的官网下载。于是修改指令：

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0

至此，安装结束。
小提示：因为安装包很多，也比较大，在安装过程中或多或少的会有几个包一次下载不下来，这时候不要直接结束终端，只需要重复输入上面的安装指令即可，已经下载好的包不会重复下载。
四，测试
pytorch官网的测试代码：

from __future__ import print_function  
import torch
x = torch.rand(5, 3)
print(x)

得到输出如图：