這篇文章是記錄了自己的學習過程,希望方便自己回憶的同時,也能夠爲其他小夥伴提供一些幫助。
使用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__)