当前位置: 首页 > news >正文

tensorflow-gpu==1.14.0 测试cuda是否成功

tensorflow-gpu==1.14.0 测试cuda是否成功

代码:


import tensorflow as tfflag = tf.test.is_gpu_available()
print(flag)



image