亲测有用
CUDA_VISIBLE_DEVICES=1 python train.py
Keras指定使用GPU
最新推荐文章于 2023-06-10 13:08:43 发布
亲测有用
CUDA_VISIBLE_DEVICES=1 python train.py