GPU版TensorFlow怎麼指定讓CPU運行

因爲某些緣由GPU版的TensorFlow運行起來會出現一些問題,好比內存溢出等狀況。此時咱們能夠用CPU和系統內存來運行咱們的程序。html 代碼以下:post import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 將以上三行代碼放入程序首部便可。 --
相關文章
相關標籤/搜索