美文网首页
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