首先若不加任何配置情況下，是默認使用gpu的，
加上下面這句代碼就使用cpu了。
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
tensorflow2.0 GPU和CPU切換
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
首先若不加任何配置情況下，是默認使用gpu的，
加上下面這句代碼就使用cpu了。
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"