更多细节请参考:
使用conda/pip安装pytorch(mac/windows/linux)
PyTorch官网教程
本文以Linux中-pytorch版本1.0.0-清华源为-cuda90为例:
Cuda官网教程
同时请注意检查依赖库安装:
pytorch安装依赖库
在安装过程中遇到的问题,会反馈更新在:
pytorch安装错误汇总
(Linux/windows/Mac OS都相近,除去源码安装的部分细节以及Mac OS不支持CUDA)
1.pip安装pytorch:
(若为python则用pip,若为python3则用pip3)
sudo pip3 install torch==1.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
安装过程中小白可以都用一下sudo,以免出现“Permission Deny”等问题,当然具体情况具体分析。
没有pip或者python请点击传送门:
pip安装教程
python安装教程
补充
也可以直接默认安装最新的pytorch的GPU版本:
sudo pip3 install torch torchvision
pip清华源升级pytorch:
sudo pip3 install --upgrade torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple
pip降级安装(以降级到torch 0.4.1为例):
方法一(简单暴力)
sudo pip3 install torch==0.4.1 -i https://pypi.tuna.tsinghua.edu.cn/simple>
方法二(容易出错)
sudo pip3 install https://download.pytorch.org/whl/cu90/torch-0.4.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision
(注意其中的cu90/torch-0.4.1-cp35-cp35m-win_amd64.whl,会随你的cuda, python, 版本需求改变)
pip卸载pytorch :
sudo pip3 uninstall torch
事实个人发现,'-i' 后的镜像网址输错一点,有时候也会更正。
2.conda 安装pytorch
cuda 90下安装pytorch:
conda install pytorch torchvision cuda90 -c soumith
pytorch 无cuda版本:
conda install pytorch-cpu torchvision-cpu -c pytorch
conda卸载pytorch
conda uninstall pytroch