使用CUDA_VISIBLE_DEVICES。此时的运行命令如下：
CUDA_VISIBLE_DEVICES=1 python train.py
Keras指定使用GPU运算
猜你喜欢
转载自blog.csdn.net/A_a_ron/article/details/78450672
今日推荐
周排行