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 红外遥控模块)

热门推荐

01GitHub 镜像站点 02如何新建文件夹？电脑新建文件夹的4种方法 032026年7月AI圈大地震：GPT-5.6被政府限制、Claude入驻Slack、Anthropic自研芯片 04国内可直接用、免费额度/永久免费的大模型API清单（含 SiliconFlow、火山、阿里、智谱、百度、Kimi、DeepSeek、DMXAPI 等）05AI科技热点日报 | 2026年07月01日 06幻兽帕鲁 - 服务器管理员权限与 GM 命令完全指南 07AI 编程 IDE 全景解析 2026：Agent 全面接管开发链路 08微信历史版本含下载地址（ Windows PC | 安卓 | MAC ）及设置微信不更新 09【解构】DeepSeek V4 发布：技术报告深度解读 + 横向对比六大开源模型，我们的判断是……102026 国产 AI 大模型横评：DeepSeek、通义千问、Kimi、文心一言、星火、豆包谁更能打？