pytorch安装及卸载

2023-03-21 20:33| 来源: 网络整理| 查看: 265

pytorch安装及卸载一、pytorch安装

我是已经安装过了gpu版本的tensorflow

1.1 先打开cmd看看自己安装的cuda版本

输入

nvcc --version

在这里插入图片描述可以看到是cuda 10.0，当然你也可以更新英伟达驱动，重新安装更高版本的cuda，使用更高版本的pytorch

1.2 找到离线下载包进行安装

pytorch离线下载地址：pytorch离线下载

根据windows系统+python版本——cuda版本进行下载在这里插入图片描述

1.3 使用pip安装 pip install torch-1.0.0-cp36-cp36m-win_amd64.whl

torch 1.0.0 对应的torchvision的版本为0.2.0

pip install torchvision==0.2.0

其他对应的版本： torch对应的torchvision版本

或者从https://pytorch.org/这里看，然后下载离线包，在这里插入图片描述 gpu pytorch对应的torchvision版本也得是GPU的。

1.4 验证gpu版本是否安装成功

检测程序如下：

import torch print(torch.__version__) print(torch.cuda.is_available())

运行结果：

1.0.0 True

PyTorch版本为1.0.0 True代表GPU版本成功安装！

二、pytorch卸载 1、使用conda卸载Pytorch conda uninstall pytorch conda uninstall libtorch 2、使用pip卸载Pytorch pip uninstall torch

如果当初是用conda install 命令安装的pytorch，则用第一种方法，如果是用pip install命令安装的，则用第二种方法。或者两种方法都试一下。

【本文地址】

公司简介

联系我们