【工程篇】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())
相关推荐
技术闲聊DD2 分钟前
机器学习(1)- 机器学习简介
人工智能·机器学习
mwq301239 分钟前
GPT-2 中的残差权重初始化
人工智能
mwq3012336 分钟前
Transformer : 深度神经网络中的残差连接 (Residual Connection)
人工智能
信田君952738 分钟前
瑞莎星瑞(Radxa Orion O6) 基于 Android OS 使用 NPU的图片模糊查找APP 开发
android·人工智能·深度学习·神经网络
StarPrayers.1 小时前
卷积神经网络(CNN)入门实践及Sequential 容器封装
人工智能·pytorch·神经网络·cnn
周末程序猿1 小时前
谈谈上下文工程(Context Engineering)
人工智能
一水鉴天1 小时前
整体设计 逻辑系统程序 之29 拼语言+ CNN 框架核心定位、三阶段程序与三种交换模式配套的方案讨论 之2
人工智能·神经网络·cnn
海森大数据1 小时前
AI破解数学界遗忘谜题:GPT-5重新发现尘封二十年的埃尔德什问题解法
人工智能·gpt
望获linux2 小时前
【实时Linux实战系列】Linux 内核的实时组调度(Real-Time Group Scheduling)
java·linux·服务器·前端·数据库·人工智能·深度学习