赞
踩
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 后为安装的环境
pip uninstall torch
或者用conda命令
conda uninstall pytorch
import torch
print(torch.__version__)
# cuda的版本查看
print(torch.version.cuda)
print(torch.cuda.amp)
print(torch.cuda.amp.autocast)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。