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

相关推荐
国科安芯4 分钟前
AS32S601型抗辐射MCU在分布式太空算力架构中的技术演进与应用前景
人工智能·分布式·单片机·嵌入式硬件·架构·边缘计算
K姐研究社6 分钟前
GLM-5.2 实测 – 代码生成能力跻身全球第一梯队
人工智能·aigc
OpenAnolis小助手10 分钟前
龙蜥社区智算联盟发布《智算集群基础设施运维通用技术要求》(附下载链接)
运维·人工智能·智算运维·龙蜥智算联盟
HySpark11 分钟前
情感识别实践(三):模型怎么选?BERT只是起点,多模态才是终点
人工智能·深度学习·bert
白执落13 分钟前
使用 Python + LangChain + Vue3 构建 LLM 聊天应用
人工智能·python·langchain
叫我Paul就好15 分钟前
RAG 从入门到精通 - 基础版本
人工智能·软件工程·rag
weixin_3975740917 分钟前
语义鸿沟是企业AI落地的真正卡点
人工智能
蓝创工坊Blue Foundry18 分钟前
本地 PDF、图片字段提取到 Excel:用文档工作台的完整流程
python·ocr·vim·paddlepaddle
硅徒19 分钟前
蓝队检测规则编写:从告警噪声里捞出真实攻击的工程方法
人工智能