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)

得到输出如图:
在这里插入图片描述

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐