背景

安装了TensorFlow,训练的时候死活用不到GPU,特别是在docker环境中,GPU是否可检查更为重要

依赖条件

已经安装好python、响应版本GPU的驱动和nvidia的驱动

检测代码

import tensorflow as tf

print(tf.__version__)
print(tf.reduce_sum(tf.random.normal([1000, 1000])))
print(tf.config.list_physical_devices('GPU'))

背景

在有GPU的windows上安装TensorFlow

依赖条件

此文文档主要基于windows平台
已经安装好GPU的驱动
已经安装好 conda 环境

参考资料

windows受支持的版本对应关系 windows受支持的版本对应关系