这篇文章是记录了自己的学习过程,希望方便自己回忆的同时,也能够为其他小伙伴提供一些帮助。
使用pip与conda安装、更新与卸载Pytorch和torchvision
一 利用pip
1.1 pip安装pytorch
#不指定版本
pip install pytorch
#安装指定版本0.4.1
pip install torch==0.4.1
提示,换源可以加快安装速度,换源可以参考pip更换国内的安装源 / 镜像
1.2 pip更新pytorch
#更新pytorch和torchvision安装包
pip install --upgrade torch torchvision
1.3 pip卸载pytorch
若上述方法无法更新pytorch,可卸载再重新安装pytorch,卸载方法如下:
pip uninstall torch
二 利用conda
2.1 conda安装pytorch
#不指定版本
conda install pytorch
#不指定版本,同时安装torchvision,有cuda
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
#安装指定版本
conda install pytorch==0.4.1
2.2 conda更新pytorch
#只更新pytorch安装包
conda update pytorch
#可以同时更新pytorch和torchvision安装包
conda update pytorch torchvision
2.3 conda卸载pytorch
conda uninstall pytorch
三 查看当前环境下pytorch版本
进入python编译环境,然后输入
import torch
print(torch.__version__)