TensorFlow测试GPU
我安装的TensorFlow是2.4.0,测试GPU是否可以调用可以用以下代码
import tensorflow as tf
print(tf.__version__)
tf.config.list_physical_devices('GPU')
另外提一句
tf.test.is_gpu_available()
这句代码虽然也可以运行,但是我这是2.4.0版,提示也说了后续要删除
还是我上面的那行就行
TensorFlow测试GPU
我安装的TensorFlow是2.4.0,测试GPU是否可以调用可以用以下代码
import tensorflow as tf
print(tf.__version__)
tf.config.list_physical_devices('GPU')
另外提一句
tf.test.is_gpu_available()
这句代码虽然也可以运行,但是我这是2.4.0版,提示也说了后续要删除
还是我上面的那行就行