当前位置:   article > 正文

Pytorch如何更新版本与卸载_pytorch升级到指定版本

pytorch升级到指定版本

今天我们主要汇总如何使用使用ubuntu,CentOS,Mac更新Pytorch和torchvision,以及如何查看当前python版本。本教程只汇集如何使用pip,conda更新以及卸载Pytorch和torchvision,希望对您有所帮助!

一、更新Pytorch和torchvision

1、使用Pip更新Pytorch和torchvision

# 列举pip当前可以更新的所有安装包
pip list --outdated --format=legacy
# 更新pytorch和torchvision安装包
pip install --upgrade pytorch torchvision
  • 1
  • 2
  • 3
  • 4

2、使用conda更新Pytorch和torchvision

# 建议将其添加soumith为您的Anaconda(或Miniconda)的源服务器
conda config --add channels soumith
# 更新pytorch和torchvision安装包
conda update pytorch torchvision
  • 1
  • 2
  • 3
  • 4

3、卸载Pytorch重新安装

pip uninstall torch
  • 1

二、如何安装指定版本Pytorch

使用conda安装指定版本

# 比如我们安装 v10.0
conda install pytorch=10.0 -c soumith
  • 1
  • 2

使用pip安装指定版本
python3版本

pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-linux_x86_64.whl
pip3 install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp37-cp37m-linux_x86_64.whl
  • 1
  • 2

三、卸载Pytorch

1、使用conda卸载Pytorch

conda uninstall pytorch
conda uninstall libtorch
  • 1
  • 2

2、使用pip卸载Pytorch

pip uninstall torch
  • 1

四、如何查看当前pytorch版本

有时候我们想要知道当前的pytorch版本,我们可以使用如下代码打印出当前的版本:

python
>>>import torch
>>>print(torch.__version__)
  • 1
  • 2
  • 3

五、查看pytorch对应的cuda版本

python
>>>import torch
>>>torch.version.cuda
  • 1
  • 2
  • 3
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/975576
推荐阅读
相关标签
  

闽ICP备14008679号