【工程篇】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())
相关推荐
TDengine (老段)2 小时前
从 ETL 到 Agentic AI:工业数据管理变革与 TDengine IDMP 的治理之道
数据库·数据仓库·人工智能·物联网·时序数据库·etl·tdengine
蓝桉8022 小时前
如何进行神经网络的模型训练(视频代码中的知识点记录)
人工智能·深度学习·神经网络
星期天要睡觉3 小时前
深度学习——数据增强(Data Augmentation)
人工智能·深度学习
南山二毛4 小时前
机器人控制器开发(导航算法——导航栈关联坐标系)
人工智能·架构·机器人
大数据张老师4 小时前
【案例】AI语音识别系统的标注分区策略
人工智能·系统架构·语音识别·架构设计·后端架构
xz2024102****4 小时前
吴恩达机器学习合集
人工智能·机器学习
anneCoder4 小时前
AI大模型应用研发工程师面试知识准备目录
人工智能·深度学习·机器学习
骑驴看星星a4 小时前
没有深度学习
人工智能·深度学习
youcans_4 小时前
【医学影像 AI】YoloCurvSeg:仅需标注一个带噪骨架即可实现血管状曲线结构分割
人工智能·yolo·计算机视觉·分割·医学影像
空白到白5 小时前
机器学习-决策树
人工智能·决策树·机器学习