pytorch基本操作:安装、下载与版本控制

更多细节请参考:
使用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

方法一(简单暴力)

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.2.0-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


3.下载源码安装pytorch

你可能感兴趣的:(pytorch)