昨天把GPU版本的tf2.0 安裝成功之后，現在所有的代碼運行居然都在gpu上跑了，
並且在對gpu使用情況沒有限制的條件下，既然gpu內存跑滿了，代碼就崩了
怎么樣才能隨心所欲的指定代碼是在cpu還是gpu呢
首先若不加任何配置情況下，是默認使用gpu的，
加上下面這句代碼就使用cpu了
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
tensorflow2.0 實現gpu和cpu切換
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。
猜您在找
tensorflow2.0 GPU和CPU 時間對比
keras & tensorflow 列出可用GPU 和 切換CPU & GPU
【tensorflow2.0】使用多GPU訓練模型
【tensorflow2.0】使用單GPU訓練模型
『TensorFlow2.0正式版教程』極簡安裝TF2.0正式版（CPU&GPU）教程
tensorflow2.0 GPU版本鏡像文件
tensorflow2.0、keras實現Attention
FastGCN論文總結及實現(Tensorflow2.0)
二.Cuda10+RTX2070+WIN10+Tensorflow2.0安裝--Tensorflow2.0 GPU安裝
tensorflow2.0——LSTM