问题:如果windows里既装了tensorflow的CPU版本和GPU版本,运行的时候如何确定跑的是哪个版本呢?

在python环境里,运行:

import tensorflow as tf
tf.test.is_gpu_available()

根据结果可以判断。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐