1. (答题法)通过accelerate config,设置多卡(GPU)使用方法
终端输入指令,生成问答页面
accelerate config
这个方法也是可以的
2. (直接修改法)通过修改default.yaml文件,设置多卡(GPU)使用方法
后面修改直接找到这个yaml文件进行修改即可
cd ~/.cache/huggingface/accelerate
vim default_config.yaml
进入vim进行修改(需要先进行accelerate才会有这个文件生成)
3 单卡(GPU)使用方法
vim default_config.yaml
进入vim进行修改
gpu_ids是all
num_processes是1
运行指令CUDA_VISIBLE_DEVICES=4 sh test_ours.sh
就可以在第四块显卡上运行了