本地部署AI大模型之PyTorch:如何使用whl文件安装PyTorch

server/2025/3/2 0:08:10/

如果想在本地安装只支持CPU的PyTorch,可以参考这篇博客。

我们需要安装支持CUDA 12.6版本的PyTorch,但是我们在直接使用官网上的指令("pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126")安装PyTorch时报了如下的错误:

因此,我们准备使用whl文件来安装PyTorch(包括torch, torchvision, torchaudio这三个package)。

要下载 PyTorch的 .whl 文件,可以按照以下方法操作:

1. PyTorch 官方网站下载

  1. 访问 PyTorch 下载页面

    1. 打开浏览器,访问https://download.pytorch.org/whl/cu126 (因为我们想下载的是支持cuda 12.6版本的Pytorch,如果想用别的版本的cuda,比如12.4的话,把前面的链接中的"cu126"改成"cu124"即可)。
  2. 选择合适的版本

    1. 根据你的 Python 版本和 CUDA 版本,选择相应的 .whl 文件。例如,我们使用的是 Python 3.11 和 CUDA 12.6,可以下载 torch-2.6.0+cu126-cp311-cp311-win_amd64.whl(Windows 系统)或 torch-2.6.0+cu126-cp311-cp311-linux_aarch64.whl(Linux 系统)。
    2. 同样,对于torchvision,根据Python版本和CUDA12.6,我们可以下载torchvision-0.21.0+cu126-cp311-cp311-win_amd64.whl(Windows 系统)或 torchvision-0.21.0+cu126-cp311-cp311-linux_x86_64.whl(Linux 系统)。
    3. 同样,对于torchaudio,根据Python版本和CUDA12.6,我们可以下载torchaudio-2.6.0+cu126-cp311-cp311-win_amd64.whl(Windows 系统)或 torchaudio-2.6.0+cu126-cp311-cp311-linux_x86_64.whl(Linux 系统)。
  3. 下载文件

    1. 点击链接下载 .whl 文件到本地。

2.安装 .whl 文件

  1. 安装 .whl 文件
    1. 下载完成后,打开命令提示符或终端,导航到 .whl 文件所在的目录,运行以下命令:

pip install <filename>.whl

将 <filename> 替换为你下载的 .whl 文件的名称。例如:

pip install torch-2.6.0+cu126-cp311-cp311-win_amd64.whl

安装完成后,可以看到下面的结果:

最后,检查一下本地是否能够使用PyTorch。我们通过在本地的Python文件中检查是否能够import torch来判断是否能够使用PyTorch:

出现上图中红框的结果的话,则表明PyTorch安装成功。


http://www.ppmy.cn/server/171658.html

相关文章

探索超声波的奥秘——定时器与PCA

超声波技术的诞生灵感来源于大自然中的回声定位现象&#xff0c;尤其是蝙蝠的独特能力。蝙蝠通过发出高频超声波并捕捉回声来精确地探测周围的物体和猎物&#xff0c;即使在漆黑的夜晚也能轻松导航。 在单片机中&#xff0c;也有着超声波这个模块&#xff0c;它在单片机上的标识…

3D Web轻量化引擎HOOPS Communicator如何赋能航空航天制造?

在当今航空航天制造领域&#xff0c;精确度、效率和协作是推动行业发展的关键要素。随着数字化技术的飞速发展&#xff0c;3D Web可视化开发包HOOPS Communicator 为航空航天制造带来了革命性的变化。它凭借强大的功能和灵活的应用&#xff0c;助力企业在设计、生产、培训等各个…

【Web 大语言模型攻击简介】

Web 大语言模型攻击简介 一、攻击原理与分类二、检测与防御技术三、典型利用方式与案例四、防御建议与未来挑战总结 关于 Web 大语言模型攻击的原理、检测及利用方式的简介&#xff1a; 一、攻击原理与分类 提示注入&#xff08;Prompt Injection&#xff09; 核心机制&#xf…

AI算力革命重塑交换机市场:国产厂商的破局路径与战略高地

一、算力需求爆炸与网络架构重构 1.1 大模型训练对交换机的极限挑战 当前AI大模型参数量已突破十万亿级别&#xff0c;单次训练需协调数万张GPU卡。以GPT-4为例&#xff0c;其训练集群需要25,000块A100显卡&#xff0c;显存带宽达2TB/s&#xff0c;传统400G交换机每秒600GB的…

如何防止 Docker 注入了恶意脚本

根据您的描述&#xff0c;攻击者通过 CentOS 7 系统中的 Docker 注入了恶意脚本&#xff0c;导致自动启动名为 “masscan” 和 “x86botnigletjsw” 的进程。这些进程可能用于网络扫描或其他恶意活动。为了解决这一问题&#xff0c;建议您采取以下步骤&#xff1a; 1. 停止并删…

测试金蝶云的OpenAPI

如何使用Postman测试K3Cloud的OpenAPI 1. 引言 在本篇博客中&#xff0c;我将带你逐步了解如何使用Postman测试和使用K3Cloud的OpenAPI。内容包括下载所需的SDK文件、配置文件、API调用及测试等步骤。让我们开始吧&#xff01; 2. 下载所需的SDK文件 2.1 获取SDK 首先&…

蓝桥杯---快速排序(leetcode第159题)最小的k个元素(剑指offer原题)

文章目录 1.题目概述2.思路分析3.代码详解 1.题目概述 这个题目只是被包装了一下&#xff0c;本质上依然是使用的我们的快速排序算法&#xff0c;为什么这样说呢&#xff1f;因为仔细阅读题目你就会发现&#xff0c;这个需要我们去找到最小的前K个元素&#xff0c;并且进行返回…

Python毕业设计选题:基于django+vue的疫情数据可视化分析系统

开发语言&#xff1a;Python框架&#xff1a;djangoPython版本&#xff1a;python3.7.7数据库&#xff1a;mysql 5.7数据库工具&#xff1a;Navicat11开发软件&#xff1a;PyCharm 系统展示 管理员登录 管理员功能界面 用户管理 员工管理 疫情信息管理 检测预约管理 检测结果…