当前位置: 首页 > news >正文

Ubuntu如何更换 PyTorch 版本

环境:

Ubuntu22.04

WLS2

问题描述:

Ubuntu如何更换 PyTorch 版本考虑安装一个为 CUDA 11.5 编译的 PyTorch 版本。如何安装旧版本

解决方案:

决定不升级CUDA版本,而是使用一个与CUDA 11.5兼容的PyTorch版本,您可以按照以下步骤来安装一个较旧版本的PyTorch。

  1. 卸载当前的PyTorch版本:首先,您需要卸载当前安装的PyTorch,以避免任何潜在的版本冲突。在命令行中运行以下命令:

    pip uninstall torch torchvision torchaudio
    

在这里插入图片描述

  1. 查找合适的PyTorch版本:接下来,您需要查找一个为CUDA 11.5编译的PyTorch版本。您可以访问PyTorch官方网站,并在“Previous PyTorch Versions”部分找到与CUDA 11.5兼容的版本。

  2. 安装兼容的PyTorch版本:在确定了哪个PyTorch版本与CUDA 11.5兼容之后,您可以使用pip或conda来安装。例如,如果您找到PyTorch版本1.9.0是与CUDA 11.5兼容的,您可以使用以下命令安装:

    使用pip的话:
    install pytorch1.12.1 torchvision0.13.1 cudatoolkit=11.3 -c pytorch

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html

这里,-f https://download.pytorch.org/whl/torch_stable.html会告诉pip使用官方的PyTorch资源库来找到与CUDA 11.5兼容的PyTorch版本。

在这里插入图片描述

请注意,您需要将`1.9.0`替换为您找到的正确的版本号,而`+cu115`表示这个包是为CUDA 11.5编译的。在运行这些命令之前,请确保您的环境(例如conda环境)是激活的。
  1. 验证安装:安装完成后,您应该验证PyTorch和CUDA是否正确安装。在Python中运行以下代码:
python -c "import torch; print(torch.__version)

在这里插入图片描述

这应该会输出PyTorch的版本,PyTorch使用的CUDA版本,以及一个布尔值,指示CUDA是否可用。

请确保在您的工作环境或虚拟环境中执行这些步骤,避免对系统级别的Python环境产生影响。


http://www.lryc.cn/news/342410.html

相关文章:

  • python flask css样式无效
  • 大数据学习笔记14-Hive基础2
  • vue3 下载图片(包括多图片下载)
  • LabVIEW如何通过子VI更改主VI控件属性?
  • 关于MS-DOS时代的回忆
  • 数据库索引(Mysql)
  • 异常-Exception
  • ctfshow——SQL注入
  • 第十三章 计算机网络
  • 商品详情 API 返回值说明
  • 层级实例化静态网格体组件:开启大量模型处理之门
  • 【网络知识】光猫、路由器 和 交换机 的作用和区别?
  • 初识Electron,创建桌面应用
  • AI编码时代到来?实现编程梦想的利器—Baidu Comate测评
  • 去中心化自治组织(DAO)
  • MySQL之多表查询
  • 极端天气频发,我们普通人如何保全自己
  • 直面市场乱价,品牌商家该如何解决?
  • Spring中的Bean相关理解
  • 操作系统实战(二)(linux+C语言)
  • 哪些情况下会触发MySQL的预读机制?
  • react使用谷歌人机验证
  • java JMH 学习
  • 本地运行AI大模型简单示例
  • 图像处理:时域、空域、频率的滤波介绍
  • TC8002D 是一颗带关断模式的音频功放IC
  • 深度学习之基于Vgg19预训练卷积神经网络图像风格迁移系统
  • MySQL:多表查询练习
  • # 从浅入深 学习 SpringCloud 微服务架构(八)Sentinel(1)
  • [微信小程序] 入门笔记2-自定义一个显示组件