https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_411.31_win10
昨天把GPU版本的tf2.0 安裝成功之后，現在所有的代碼運行居然都在gpu上跑了，
並且在對gpu使用情況沒有限制的條件下，既然gpu內存跑滿了，代碼就崩了
怎么樣才能隨心所欲的指定代碼是在cpu還是gpu呢
首先若不加任何配置情況下，是默認使用gpu的，
加上下面這句代碼就使用cpu了
|
1
|
os.environ[
"CUDA_VISIBLE_DEVICES"
]
=
"-1"
|
https://blog.csdn.net/qq_35148758/article/details/93785285
指定GPU編號
import os os.environ["CUDA_VISIBLE_DEVICES"] = "0" 或者在腳本或者命令行中指定 export CUDA_VISIBLE_DEVICES=０