【工程篇】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())
相关推荐
limingade16 分钟前
手机打电话时如何识别对方按下的DTMF按键的字符-安卓AI电话机器人
android·人工智能·物联网·智能手机·语音识别·信号处理
紫雾凌寒1 小时前
计算机视觉 |解锁视频理解三剑客——SlowFast
人工智能·python·深度学习·计算机视觉·音视频·视频理解·slowfast
说私域1 小时前
基于定制开发开源AI大模型S2B2C商城小程序的商品选品策略研究
人工智能·小程序·开源
说私域1 小时前
价格锚定策略与定制开发开源链动2+1模式S2B2C商城小程序的应用探索
人工智能·小程序·开源·流量运营
赛逸展张胜1 小时前
CES Asia 2025将聚焦AI与自动驾驶,行业先锋共探未来出行变革
人工智能·机器学习·自动驾驶
泰迪智能科技011 小时前
企业数据挖掘建模平台哪家好?
人工智能·数据挖掘
sauTCc1 小时前
Pytorch加载数据的Dateset类和DataLoader类
人工智能·pytorch·python
啥都鼓捣的小yao2 小时前
课程2. 用PyTorch训练神经网络与梯度下降
人工智能·pytorch·神经网络
PeterClerk2 小时前
DeepSeek R1 训练策略4个阶段解析
人工智能·深度学习·机器学习·语言模型·自然语言处理·llm·deepseek
龚大龙3 小时前
机器学习(李宏毅)——Diffusion Model
人工智能·机器学习