Keras指定使用GPU运算

来源:互联网 发布:java面向对象编程继承 编辑:程序博客网 时间:2024/06/06 07:07

使用CUDA_VISIBLE_DEVICES。此时的运行命令如下:

CUDA_VISIBLE_DEVICES=1 python train.py