昨天把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