conda安装Pytorch下载过慢解决办法以及在ubuntu下pytorch1.1安装方法)

pytorch最近已经更新到了稳定版本的1.0.1,从Pytorch官网上可以看到有多种安装方式:

比较常见的方法是通过pip和conda安装,当使用conda安装的时候,可能会遇到下载过慢的问题,尤其是文件:pytorch-1.0.1-py3.6_cuda90_cudnn7_1.tar.bz2,下载的时候可能会遇到无尽的等待。这里推荐用清华源替代默认conda源的方法,可以解决下载过慢的问题。

清华conda源地址:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

添加清华源


   
   
   
   
  1. conda config -- add channels https: //mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  2. conda config -- add channels https: //mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  3. conda config -- set show_channel_urls yes

另外为了保险起见,建议同时添加第三方conda源:


   
   
   
   
  1. conda config -- add channels https: //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  2. conda config -- add channels https: //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  3. conda config -- add channels https: //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  4. conda config -- add channels https: //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  5. conda config -- add channels https: //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

安装PyTorch

根据Python和CUDA选择对应的版本,然后官方给出提示可通过运行:

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
但是这里一定要注意,去掉-c pytorch,安装的时候才会默认从清华源下载相应的包,因此这里用命令行:

conda install pytorch torchvision cudatoolkit=9.0
   
   
   
   

接着等待安装成功就好了(注意到画红框的部分已经默认源替换为tsinghua源了)。

最后可以通过命令行验证一下,可以看到已经正确安装PyTorch1.0.1。

--------------------------------------------------------------------------------------------------(这是一条华丽的分割线)

3月5日更新ubuntu下pytorch1.0.1安装方法(Ubuntu16.04+CUDA9.0+PyTorch1.0.1)

今天在ubuntu下安装pytorch1.0.1的时候发现,清华源竟然坑爹的没有cudatoolkit9.0的包。

于是,尝试了一下发现中科大源中有ubuntu下cuda9.0的包,因此这样配置conda 源:

sudo gedit ~/.condarc
   
   
   
   

配置文件修改如下:


   
   
   
   
  1. channels:
  2. - https: //mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  3. - https: //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  4. - https: //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  5. - https: //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  6. - https: //mirrors.ustc.edu.cn/anaconda/pkgs/main/
  7. show_channel_urls: true

运行conda install pytorch torchvision cudatoolkit=9.0

成功!!!

7月23日更新ubuntu下pytorch1.1安装方法(通过pip)

通过pip安装pytorch1.1非常简单。因为清华源中没有Pytorch1.1的安装包目标,所以需要下载好pytorch1.1的whl文件,可以去官网下载,或者从这里下载cuda10.0,Python3.6的安装包,提取码nb2k。

下载好之后同样我们需要用pip清华源安装whl文件就好了,这样相关依赖环境会安装的很快:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch-1.1.0-cp36-cp36m-manylinux1_x86_64.whl
   
   
   
   

安装成功!

你可能感兴趣的:(深度学习及Python相关知识)