ubuntu中pytorch安装速度太慢的解决方法
一,电脑配置
- ubuntu_18.04
- cuda9.0
- gpu
- 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)
得到输出如图: