ubuntu24下python3.9安装pytorch

是小峰呀2024-09-01 9:23

安装gpu版pytorch

1.去官网查找命令

bash 复制代码

https://pytorch.org/get-started/previous-versions/

由于我们安装的是CUDA12.1，所以我们选择12.1的进行安装

安装完成之后通过下面代码进行测试

bash 复制代码

>>> import torch
>>> print(torch.cuda.is_available())
True

显示True表明torch可以调用GPU进行计算，说明安装成功

上一篇：Android CCodec Codec2 （五）C2Param - Ⅲ

下一篇：Linux驱动开发基础(IRDA 红外遥控模块)

热门推荐

012026年7月AI圈大地震：GPT-5.6被政府限制、Claude入驻Slack、Anthropic自研芯片 02GitHub 镜像站点 03幻兽帕鲁 - 服务器管理员权限与 GM 命令完全指南 04如何新建文件夹？电脑新建文件夹的4种方法 05AI科技热点日报 | 2026年07月01日 06国内可直接用、免费额度/永久免费的大模型API清单（含 SiliconFlow、火山、阿里、智谱、百度、Kimi、DeepSeek、DMXAPI 等）07【AI前线观察】2026年国产开源大模型全面横评：从 DeepSeek V4 到 Kimi K3，谁才是开发者的最优选择？08AI 编程 IDE 全景解析 2026：Agent 全面接管开发链路 09全面体验 Grok API 中转站（2025 · Grok 4 系列最新版）102026 国产 AI 大模型横评：DeepSeek、通义千问、Kimi、文心一言、星火、豆包谁更能打？