Pytorch的更新、卸载、版本查看

更新Pytorch和torchvision

pip更新torch和torchvision

win + R ————>  cmd
activate *******        # 进入想要进的环境（激活，比如pytorch_yolo或tensorflow）


# 列举pip当前可以更新的所有安装包
pip list --outdated --format=legacy
# 更新torch和torchvision安装包
pip install --upgrade torch torchvision

或者用conda命令
conda update pytorch torchvision

pip install --upgrade torch torchvision 会下载完整的安装包，时常较久。

安装指定版本

pip install pytorch==1.5.1

conda install pytorch=0.1.10 -c pytorch_test
conda install pytorch==1.5.1 torchvision==0.7.0 cudatoolkit=10.0 -c pytorch

-c 后为安装的环境

卸载torch

pip uninstall torch

或者用conda命令
conda uninstall pytorch

查看torch和torchvision的版本

import torch

print(torch.__version__)

# cuda的版本查看
print(torch.version.cuda)
print(torch.cuda.amp)
print(torch.cuda.amp.autocast)