在代码的开头，所有 import 之前，加入下面两行代码：
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # 这一行注释掉就是使用cpu，不注释就是使用gpu。
在代码的开头，所有 import 之前，加入下面两行代码：
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # 这一行注释掉就是使用cpu，不注释就是使用gpu。
本文标题：【python】Keras、Tensorflow 如何切换 GP
本文链接：https://www.haomeiwen.com/subject/etrlrqtx.html
网友评论