判断tf2.0以上是否能用gpu的命令

2020-05-13  本文已影响0人  胜负55开

import之后,直接运行这条命名:返回是True即当前tf是可以使用gpu的版本,若返回False则当前版本不能使用gpu!

import tensorflow as tf

tf.test.is_gpu_available()
上一篇 下一篇

猜你喜欢

热点阅读