import tensorflow as tf
from tensorflow.python.client import device_lib
print("版本:", tf.__version__)
print("型号:", device_lib.list_local_devices())
查看自己的tensorflow的版本和型号(cpu or gpu)
阅读 147
2022-11-16
import tensorflow as tf
from tensorflow.python.client import device_lib
print("版本:", tf.__version__)
print("型号:", device_lib.list_local_devices())
相关推荐
精彩评论(0)