如果有必要,记得先创建一个anaconda3虚拟环境:
onda create --name 环境名 python==3.7
cuda:
wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
sudo sh cuda_11.0.2_450.51.05_linux.run
PyTorch
豆瓣源:
pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple some-package
或者1.7.0最新稳定版本:
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple some-package
pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
TensorFlow
pip install --index-url https://pypi.douban.com/simple tensorflow
或指定版本:
install --index-url https://pypi.douban.com/simple tensorflow==1.15.0
TF-gpu2.X
pip install --index-url https://pypi.douban.com/simple tensorflow-gpu==2
win10+cuda10.2安装pytorch1.7.1(使用清华镜像源)
(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/
备注:
(1)torch和torchvison下载地址
https://download.pytorch.org/whl/torch_stable.html
(2)安装相应的pytorch版本
conda install pytorch torchvision cudatoolkit=10.2
(版本号从https://pytorch.org/上,查找对应版本)
参考文献:
https://blog.csdn.net/liubing8609/article/details/113359885
解决其他慢的参考文献:
https://www.zhihu.com/question/435340249
其他详细教程:
配置清华镜像源
我们使用Anaconda环境进行Pytorch的安装
在命令行中依次输入一下命令并执行
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
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/pytorch/
设置搜索时显示通道地址
conda config --set show_channel_urls yes
安装Pytorch
使用清华镜像源进行Pytorch的安装时,官方给出的安装命令就失效了
官方给出的安装命令是:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
如果使用清华镜像源安装Pytorch时,需要输入以下命令:
conda install pytorch torchvision cudatoolkit=10.1
在Linux上使用shell脚本一键安装
使用Linux系统的时候,有时一步一步复制粘贴命令非常麻烦,特别是使用远程服务器的时候。这时候就需要shell脚本一键解决复制粘贴命令的麻烦。
在任意文件目录下新建一个shell脚本文件,这里取名为 pytorch_install.sh
shell脚本文件的内容如下所示
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
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/pytorch/
设置搜索时显示通道地址
conda config --set show_channel_urls yes
安装pytorch
conda install pytorch torchvision cudatoolkit=10.1
在命令行中执行shell脚本,输入命令:
bash pytorch_install.sh
1
等待安装完成即可。