GPU版TensorFlow怎麼指定讓CPU執行

NO IMAGE

由於某些原因GPU版的TensorFlow執行起來會出現一些問題,比如記憶體溢位等情況。此時我們可以用CPU和系統記憶體來執行我們的程式。

程式碼如下:


import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

將以上三行程式碼放入程式首部即可。