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=０