目录python
添加清华源linux
安装PyTorchubuntu
3月5日更新ubuntu下pytorch1.0.1安装方法(Ubuntu16.04+CUDA9.0+PyTorch1.0.1)bash
7月23日更新ubuntu下pytorch1.1安装方法(经过pip)url
11月26日更新ubuntu下pytorch1.3安装(经过conda)spa
pytorch最近已经更新到了稳定版本的1.0.1,从Pytorch官网上能够看到有多种安装方式:命令行
比较常见的方法是经过pip和conda安装,当使用conda安装的时候,可能会遇到下载过慢的问题,尤为是文件:pytorch-1.0.1-py3.6_cuda90_cudnn7_1.tar.bz2,下载的时候可能会遇到无尽的等待。这里推荐用清华源替代默认conda源的方法,能够解决下载过慢的问题。code
清华conda源地址:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/blog
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源:ip
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/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。
--------------------------------------------------------------------------------------------------(这是一条华丽的分割线)
今天在ubuntu下安装pytorch1.0.1的时候发现,清华源居然坑爹的没有cudatoolkit9.0的包。
因而,尝试了一下发现中科大源中有ubuntu下cuda9.0的包,所以这样配置conda 源:
sudo gedit ~/.condarc
配置文件修改以下:
channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ - https://mirrors.ustc.edu.cn/anaconda/pkgs/main/ show_channel_urls: true
运行conda install pytorch torchvision cudatoolkit=9.0
成功!!!
经过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
安装成功!
首先要说明几点注意事项:
也就是说配置conda源的时候不要再加科大源了。
channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge show_channel_urls: true
这样配置就能够了。
而后有人问什么配置了源仍是下载速度慢,首先你要清楚速度慢的缘由在哪里:
整个安装耗时主要在cudatoolkit和Pytorch安装包上了,只要能在conda源中找到这两个包,就能解决速度慢的问题。
为何配置了速度还慢,固然是源里面没有对应你须要版本的cudatoolkit和pytorch安装包了。具体有没有对应安装包本身去下面的连接里查一下:
顺便注意一下,源里面的cuda版本和python版本以及pytorch版本都有对应关系,不是全部版本的pytorch+cuda+python都能随便装。
以及我帮你们查了一下,截至到今天,清华源支持的cudatoolkit有:
9.0/9.2/10.0/10.1/10.2,不支持10.2。
同理,目前支持的pytorch安装包版本支持从pytorch1.3.1 + python3.7如下 + cuda9.0/9.1/10.0/10.1(仍是不支持10.2)
也就是说若是你想用清华源安装请暂时避开10.2这个坑。而且注意如下本身电脑上的cuda版本和python版本是否包括在清华源当中。
可是将来支持更多版本好比cuda10.2也是有可能的,到时候须要请本身去安装包地址查一下。
---------碎碎念分割线-------------------------------------------------------------------------------------------------------------------
言归正传,说一下我验证过的conda安装pytorch1.3 + cuda10.1 + python3.6安装成功。具体操做方法仍是老样子:
配置源:
sudo gedit ~/.condarc
配置文件修改成:
channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge show_channel_urls: true
命令行安装(注意必定去掉-c):
conda install pytorch torchvision cudatoolkit=10.1
成功!!