📌  相关文章
📜  获取 gpu 设备名称 tensorflow - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:48.657000             🧑  作者: Mango

代码示例2
import tensorflow as tf

tf.test.gpu_device_name()

# Bash
# $ CUDNN_H_PATH=$(whereis cudnn.h)
#$ cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR