判断CUDA是否可用:

tf.test.is_built_with_cuda()

猜你喜欢

转载自www.cnblogs.com/wal1317-59/p/12652487.html