檢測自己tensorflow版本是使用CPU還是GPU

打開命令界面或者pycharm中輸入: python import tensorflow as tf sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) 這裏可以看到是cpu支持所以是cpu,如果顯示gpu那就是gpu的
相關文章
相關標籤/搜索