keras 指定程序在某块卡上训练实例

更新时间：2020年06月22日 10:08:57 作者：BYR_jiandong

这篇文章主要介绍了keras 指定程序在某块卡上训练实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

场景：某台机器上有三块卡，想同时开三个程序，放到三块卡上去训练。

策略：CUDA_VISIBLE_DEVICES=1 python train.py就可以指定程序在某块卡上训练。

补充知识：keras指定GPU及显存使用量

指定GPU

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

指定GPU和显存使用量

import os
from keras.backend.tensorflow_backend import set_session

os.environ["CUDA_VISIBLE_DEVICES"] = "0"
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3
set_session(tf.Session(config=config))

指定GPU显存使用按需分配

import keras.backend.tensorflow_backend as KTF
import os

os.environ["CUDA_VISIBLE_DEVICES"] = "0"
config = tf.ConfigProto()
config.gpu_options.allow_growth=True 
sess = tf.Session(config=config)
KTF.set_session(sess)

以上这篇keras 指定程序在某块卡上训练实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

keras
训练

python破解WiFi教程代码,Python蹭网原理讲解
用Python生成一个简单的密码本，一般是有数字、字母和符号组成，这里用到的思路主要是穷举法。通过使用pywifi 模块，根据密码本暴力破解WiFi。本文只是从技术的角度来阐述学习Pywifi库！并不建议大家做任何破坏性的操作和任何不当的行为！
2023-01-01
numpy自动生成数组详解
这篇文章主要介绍了numpy自动生成数组详解，具有一定借鉴价值，需要的朋友可以参考下。
2017-12-12
教你用python3根据关键词爬取百度百科的内容
这篇文章介绍的是利用python3根据关键词爬取百度百科的内容，注意本文用的是python3版本以及根据关键词爬取，爬取也只是单纯的爬网页信息，有需要的可以参考借鉴。
2016-08-08
python使用pygame模块实现坦克大战游戏
这篇文章主要为大家详细介绍了python使用pygame模块实现坦克大战游戏，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2019-05-05
F.conv2d pytorch卷积计算方式
这篇文章主要介绍了F.conv2d pytorch卷积计算方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2023-02-02
python高手之路python处理excel文件(方法汇总)
用python来自动生成excel数据文件。python处理excel文件主要是第三方模块库xlrd、xlwt、xluntils和pyExcelerator，除此之外，python处理excel还可以用win32com和openpyxl模块
2016-01-01
Python的Django REST框架中的序列化及请求和返回
这篇文章主要介绍了Python的Django REST框架中的序列化及请求和返回,使用Django REST来实现RESTful web service非常方便和强大,需要的朋友可以参考下
2016-04-04
Django 前后台的数据传递的方法
本篇文章主要介绍了Django 前后台的数据传递的方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2017-08-08
Kears+Opencv实现简单人脸识别
这篇文章主要为大家详细介绍了Kears+Opencv实现简单人脸识别，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2019-08-08
Swin Transformer图像处理深度学习模型
这篇文章主要为大家介绍了Swin Transformer图像处理深度学习模型详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2023-03-03