作者你好,在accelerate_train.py中看到默认使用的是当前全部可用的GPU,怎么设置使其只用单个GPU呢?
作者你好,在accelerate_train.py中看到默认使用的是当前全部可用的GPU,怎么设置使其只用单个GPU呢?