【工程篇】01:GPU可用测试代码

pytorch

python 复制代码
>>> import torch
>>> print(torch.__version__)
>>> print(torch.cuda.is_available())

tensorflow

python 复制代码
>>> import tensorflow as tf
>>> print(tf.test.is_gpu_available())
相关推荐
Coding茶水间35 分钟前
基于深度学习的安全帽检测系统演示与介绍(YOLOv12/v11/v8/v5模型+Pyqt5界面+训练代码+数据集)
图像处理·人工智能·深度学习·yolo·目标检测·计算机视觉
weixin79893765432...1 小时前
Vue + Express + DeepSeek 实现一个简单的对话式 AI 应用
vue.js·人工智能·express
nju_spy1 小时前
ToT与ReAct:突破大模型推理能力瓶颈
人工智能·大模型·大模型推理·tot思维树·react推理行动·人工智能决策·ai推理引擎
AI-智能1 小时前
别啃文档了!3 分钟带小白跑完 Dify 全链路:从 0 到第一个 AI 工作流
人工智能·python·自然语言处理·llm·embedding·agent·rag
y***86692 小时前
C机器学习.NET生态库应用
人工智能·机器学习
deng12042 小时前
基于LeNet-5的图像分类小结
人工智能·分类·数据挖掘
OpenAnolis小助手2 小时前
直播预告:LLM for AIOPS,是泡沫还是银弹? |《AI 进化论》第六期
人工智能
我一身正气怎能输3 小时前
游戏大厂A*寻路优化秘籍:流畅不卡顿
人工智能·游戏
johnny2334 小时前
AI工作流编排平台
人工智能
百***35484 小时前
DeepSeek在情感分析中的细粒度识别
人工智能