Ubuntu如何更换 PyTorch 版本

devtools/2024/10/19 6:21:46/

环境:

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.ppmy.cn/devtools/26006.html

相关文章

win11 安装qt5.14.2 、qtcreator、vs编译器 。用最小安装进行 c++开发qt界面

系统 :win11 一、安装vs生成工具 ,安装编译器 下载visualstudio tools 生成工具: 安装编译器 和 windows sdk: 安装debug 调试器: 二、Qt5.14.2下载 下载链接: Index of /archive/qt/5.14/5.14.2 安装qt 三、配置QT/…

uni-app如何监测获取页面视图出现

在 uni-app 中,监测页面视图的出现或渲染完成,可以使用生命周期函数和一些自定义方法。这里有一些常见的方法: 使用生命周期函数: uni-app 提供了与页面生命周期相关的函数,如 onLoad、onShow、onReady 等。 onLoad…

leetcode_37.解数独

37. 解数独 题目描述:编写一个程序,通过填充空格来解决数独问题。 数独的解法需 遵循如下规则: 数字 1-9 在每一行只能出现一次。数字 1-9 在每一列只能出现一次。数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。(请参考…

基于STM32的环境检测系统与仿真

基于STM32的环境检测系统与仿真 1、OLED显示温湿度、光照强度以及CO浓度 2、按键调整阈值。超限之后相应LED灯亮起蜂鸣器鸣叫 3、WiFi模块显示不同数值、阈值以及当前设定值 总结 项目本身还是比较好的。 禁止转载!!!

吐槽3家知名的AI智能体

关注卢松松,会经常给你分享一些我的经验和观点。 我花了2天时间,把松松最近1年的爆款文案关键词情绪口头禅整理出来,4000多字的Prompt,都是一点点打出来的,再投喂到AI大模型里。使用的平台包括:通义千问、…

28377d升级

dsp 28377在线升级 实例总结_f021_cpu0_w1_register_address-CSDN博客

十大USDT交易平台大全XEX交易所

USDT是一种基于比特币区块链网络的加密代币,主要运用于数字货币交易平台,以稳定币为主。USDT的核心价值在于其与真实货币的固定兑换比率1:1,所以被称为Tether。随着加密货币市场的不断壮大,越来越多的交易平台开始支持USDT&#x…

使用groovy+spock优雅的进行单测

使用groovyspock优雅的进行单测 1. groovyspock示例1.1 简单示例1.2 增加where块的示例1.3 实际应用的示例 2. 单测相关问题2.1 与SpringBoot融合2.2 单测数据与测试数据隔离2.3 SQL自动转换(MySQL -> H2) 参考 Groovy是一种基于JVM的动态语言&#x…