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

相关推荐
AndrewHZ6 分钟前
【图像处理基石】如何入门色彩评估?
图像处理·人工智能·深度学习·色彩科学·hvs·色彩评估·颜色工程
TomatoSCI6 分钟前
聚类的可视化选择:PCA / t-SNE丨TomatoSCI分析日记
人工智能·机器学习
大咖分享课8 分钟前
深度剖析:最新发布的ChatGPT Agent 技术架构与应用场景
人工智能·openai·智能助手·ai代理·chatgpt agent·自主任务执行
chao_78916 分钟前
更灵活方便的初始化、清除方法——fixture【pytest】
服务器·自动化测试·python·pytest
lucky_lyovo18 分钟前
卷积神经网络--网络性能提升
人工智能·神经网络·cnn
liliangcsdn22 分钟前
smolagents - 如何在mac用agents做简单算术题
人工智能·macos·prompt
nju_spy26 分钟前
周志华《机器学习导论》第8章 集成学习 Ensemble Learning
人工智能·随机森林·机器学习·集成学习·boosting·bagging·南京大学
心情好的小球藻1 小时前
Python应用进阶DAY9--类型注解Type Hinting
开发语言·python
都叫我大帅哥1 小时前
LangChain加载HTML内容全攻略:从入门到精通
python·langchain
静心问道1 小时前
TrOCR: 基于Transformer的光学字符识别方法,使用预训练模型
人工智能·深度学习·transformer·多模态