使用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__)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章