对于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 |