ubuntu24下python3.9安装pytorch

安装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进行计算,说明安装成功

相关推荐
快起来别睡了8 分钟前
LangChain 介绍及使用指南:从“会聊天”到“能干活”的 AI 应用开发工具
人工智能
AI数据皮皮侠18 分钟前
中国区域10m空间分辨率楼高数据集(全国/分省/分市/免费数据)
大数据·人工智能·机器学习·分类·业界资讯
静心问道26 分钟前
大语言模型能够理解并可以通过情绪刺激进行增强
人工智能·语言模型·大模型
站大爷IP27 分钟前
Python文件操作的"保险箱":with语句深度实战指南
python
运器12333 分钟前
【一起来学AI大模型】算法核心:数组/哈希表/树/排序/动态规划(LeetCode精练)
开发语言·人工智能·python·算法·ai·散列表·ai编程
aneasystone本尊36 分钟前
管理 Claude Code 的工具权限
人工智能
聚客AI1 小时前
大模型学习进阶路线图:从Prompt到预训练的四阶段全景解析
人工智能·llm·掘金·日新计划
晓13131 小时前
第七章 OpenCV篇——角点检测与特征检测
人工智能·深度学习·计算机视觉
DeepSeek大模型官方教程1 小时前
NLP之文本纠错开源大模型:兼看语音大模型总结
大数据·人工智能·ai·自然语言处理·大模型·产品经理·大模型学习
MidJourney中文版2 小时前
深度报告:中老年AI陪伴机器人需求分析
人工智能·机器人