GPU版TensorFlow怎么指定让CPU运行
生活随笔
收集整理的這篇文章主要介紹了
GPU版TensorFlow怎么指定让CPU运行
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
由于某些原因GPU版的TensorFlow運行起來會出現一些問題,比如內存溢出等情況。此時我們可以用CPU和系統內存來運行我們的程序。
代碼如下:
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
將以上三行代碼放入程序首部即可。
?
總結
以上是生活随笔為你收集整理的GPU版TensorFlow怎么指定让CPU运行的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: GPUtil是一个Python模块,使用
- 下一篇: 禁用GPU版本TensorFlow,切换