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环境产生影响。


相关推荐

  1. Ubuntu1804下如何切换python版本

    2024-05-01 06:46:02       32 阅读
  2. Mac如何更换终端默认的python版本

    2024-05-01 06:46:02       13 阅读
  3. 更加适合CUDA11.2的pytorch显卡版本

    2024-05-01 06:46:02       30 阅读

最近更新

  1. TCP协议是安全的吗?

    2024-05-01 06:46:02       16 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-05-01 06:46:02       16 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-05-01 06:46:02       15 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-05-01 06:46:02       18 阅读

热门阅读

  1. macOS asdf 工具版本管理器

    2024-05-01 06:46:02       9 阅读
  2. 云计算知识点-03

    2024-05-01 06:46:02       9 阅读
  3. 将SSH密钥添加到GitHub账户

    2024-05-01 06:46:02       9 阅读
  4. -CS3342

    2024-05-01 06:46:02       6 阅读
  5. Spring Boot使用Logback将某些日志输出到单独的文件

    2024-05-01 06:46:02       10 阅读
  6. Kappa系数-评估分类算法的表现

    2024-05-01 06:46:02       11 阅读
  7. Spring boot 应用引入 Spring cloud alibaba nacos

    2024-05-01 06:46:02       11 阅读
  8. NLP Step by Step -- 如何微调一个模型(1)

    2024-05-01 06:46:02       10 阅读
  9. NLP中常见的tokenize方式及token类型

    2024-05-01 06:46:02       9 阅读
  10. spring源码分析之上下文构建

    2024-05-01 06:46:02       11 阅读
  11. 2024年华东杯数学建模思路+论文+代码

    2024-05-01 06:46:02       11 阅读
  12. 自然语言处理(NLP)简介

    2024-05-01 06:46:02       10 阅读
  13. np.concatenate在图像处理中的使用

    2024-05-01 06:46:02       9 阅读