首页 > 分享 > 强制使用GPU

强制使用GPU

Tensorflow

对于Tensorflow,声明Session的时候加入device_count={‘gpu’:0}即可

import tensorflow as tf sess = tf.Session(config=tf.ConfigProto(device_count={'gpu':0})) 123 Keras

对于Keras,则调用后端函数,设置其使用如上定义的Session即可

import tensorflow as tf import keras.backend.tensorflow_backend as KTF KTF.set_session(tf.Session(config=tf.ConfigProto(device_count={'gpu':0}))) 1234 直接在终端运行时加入相关语句实现指定GPU的使用

export CUDA_VISIBLE_DEVICES=0 python test.py # 运行test.py文件时,使用编号为0的GPU卡 export CUDA_VISIBLE_DEVICES=0,2 python test.py # 运行test.py文件时,使用编号为0和2的GPU卡 1234 在程序中使用os

import os # 使用第一张与第三张GPU卡 os.environ["CUDA_VISIBLE_DEVICES"] = "0, 2" 123

相关知识

兽医人员必知! 关于动物强制免疫法规解答
浅谈宠物第三者强制责任保险
强制宠物责任险,可行吗?
饲养宠物者应做好强制免疫
2022年动物强制免疫政策依据
宠物犬强制免疫有望写入法规
农业农村部关于印发《国家动物疫病强制免疫指导意见(2022—2025年)》的通知
乌鲁木齐:上万只宠物狗开始狂犬病强制免疫
鹿城启动宠物强制免疫
兰州宠物狗将强制免疫 三县五区疫苗接种点公布

网址: 强制使用GPU https://m.mcbbbk.com/newsview142246.html

所属分类:萌宠日常
上一篇: Java StringBuild
下一篇: 找出数组a[]中符合a[i]+a