判断tf2.0以上是否能用gpu的命令
2020-05-13 本文已影响0人
胜负55开
import之后,直接运行这条命名:返回是True即当前tf是可以使用gpu的版本,若返回False则当前版本不能使用gpu!
import tensorflow as tf
tf.test.is_gpu_available()
import之后,直接运行这条命名:返回是True即当前tf是可以使用gpu的版本,若返回False则当前版本不能使用gpu!
import tensorflow as tf
tf.test.is_gpu_available()