Keras 2.1.6 (with Tensorflow 1.12.0 and CUDA 10.2) does not find GPU devices

前端 未结 0 821
谎友^
谎友^ 2021-01-07 15:04

I\'m trying to use multi-gpus with Keras as follows:

from keras.utils import multi_gpu_model

parallel_model = multi_gpu_model(model, gpus = args.gpu_num)


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题