美文网首页
keras(tensorflow)使用gpu

keras(tensorflow)使用gpu

作者: panjinya | 来源:发表于2020-01-07 14:29 被阅读0次

参考文章keras & tensorflow 列出可用GPU 和 切换CPU & GPU

可能sklearn里面的一堆自带文本分类算法,并不能用到GPU???😓反正没有实验成功

安装:
pip/conda install tensorflow-gpu==1.14.0
pip/conda install keras

验证:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
from keras import backend as K
K.tensorflow_backend._get_available_gpus()

使用:
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0" GPUid
os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 为使用CPU

相关文章

网友评论

      本文标题:keras(tensorflow)使用gpu

      本文链接:https://www.haomeiwen.com/subject/duvractx.html