新版tensorflow-gpu, 2.2.0 版推荐的测试程序与之前的1.2.0 不同,记录一下。
- Tensorflow-gpu 1.x.x, 如Tensorflow-gpu 1.2.0, 可使用以下代码
import tensorflow as tf
tf.test.is_gpu_available()
(1) jupyter notebook示例

(2)pycharm示例

- Tensoeflow-gpu 2.x.x,如Tensorflow-gpu 2.2.0, 可使用以下代码
import tensorflow as tf
tf.config.list_physical_devices('GPU')
(1) jupyter notebook

(2) pycharm 测试示例

网友评论