【笔记】更新Pytorch和torchvision,以及如何查看当前python版本

您所在的位置:网站首页 如何查看自己的gpu版本 【笔记】更新Pytorch和torchvision,以及如何查看当前python版本

【笔记】更新Pytorch和torchvision,以及如何查看当前python版本

2024-07-11 21:07| 来源: 网络整理| 查看: 265

最近Pytorch从v0.1.12更新到了v0.2/v0.3,支持了很多方法,如果大家想知道详细内容可以去Pytorch v0.2版本发布--高阶梯度,分布式PyTorch,广播,高级索引,新图层等 查看,以及Pytorch v0.3.0版本发布--pytorch性能优化提速,支持ONNX,高阶梯度以及SparseAdam优化器,好的,我们今天主要讲解如何升级Pytorch以及如何卸载Pytorch。

一、更新Pytorch和torchvision

我们主要讲解如何所以pip,conda更新Pytorch和torchvision,这样你就可以使用ubuntu,CentOS,Mac更新Pytorch和torchvision

1、使用Pip更新Pytorch和torchvision

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

2、使用conda更新Pytorch和torchvision

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

3、卸载Pytorch重新安装

如果上面的方法无法更新pytorch,您可以卸载再重新安装pytorch,卸载方法如下:

pip uninstall torch

安装可以参考Pytorch如何安装,Linux的安装Pytorch,OSX安装Pytorch教程

二、如何安装指定版本Pytorch

使用conda安装指定版本

# 比如我们安装 v0.1.10 conda install pytorch=0.1.10 -c soumith

使用pip安装指定版本

pip install pytorch==0.1.10 三、卸载Pytorch

1、使用conda卸载Pytorch

conda uninstall pytorch conda uninstall libtorch

2、使用pip卸载Pytorch

pip uninstall torch 四、如何查看当前pytorch版本

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

import torch print(torch.__version__)


【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3