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