keras 调用gpu方法

os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"gpu_options = tf.GPUOptions(allow_growth=True)sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

nice_evil

18953人浏览 · 2017-10-17 15:36:55

nice_evil  ·  2017-10-17 15:36:55 发布

如果linux服务器中keras 没有默认gpu操作的话，那么在代码前面加入这三行命令即可选择调用的gpu：

os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"（其中0.1是选择所调用的gpu）

gpu_options = tf.GPUOptions(allow_growth=True)

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

九章云极普惠算力

加入社区

更多推荐

• · AI核心知识68——大语言模型之NSP （简洁且通俗易懂版）
• · 大语言模型系列(3): Qwen2.5-VL-3B 多模态模型端侧部署
• · 基于深度学习YOLOv12的垃圾分类识别检测系统（YOLOv12+YOLO数据集+UI界面+登录注册界面+Python项目源码+模型）

AI核心知识68——大语言模型之NSP （简洁且通俗易懂版）

九章云极普惠算力

大语言模型系列(3): Qwen2.5-VL-3B 多模态模型端侧部署

九章云极普惠算力

基于深度学习YOLOv12的垃圾分类识别检测系统（YOLOv12+YOLO数据集+UI界面+登录注册界面+Python项目源码+模型）

九章云极普惠算力

欢迎加入社区

取消 确定