stable-diffusion-webui指定GPU，device_stable diffusion webui 指定gpu

stable diffusion webui 版本是1.7（其他版本也可以通过这个方法找到）

首先打开webui.py

通过这里导入可以看的是在modules 的shared_cmd_options中导入

打开shared_cmd_options 文件

找到了熟悉的cmd_args.parser，看他的导入是在cmd_args 文件中 from modules import cmd_args

import os

import launch
from modules import cmd_args, script_loading
from modules.paths_internal import models_path, script_path, data_path, sd_configs_path, sd_default_config, sd_model_file, default_sd_model_file, extensions_dir, extensions_builtin_dir  # noqa: F401

parser = cmd_args.parser
1
2
3
4
5
6
7

打开cmd_args

import argparse
import json
import os
from modules.paths_internal import models_path, script_path, data_path, extensions_dir, extensions_builtin_dir, sd_default_config, sd_model_file  # noqa: F401

parser = argparse.ArgumentParser()
...
1
2
3
4
5
6
7

有很多参数，直接在文件中搜索‘gpu’ ， ‘device’ ， ‘cuda’ 看看那个是指定gpu的参数

parser.add_argument("--device-id", type=str, help="Select the default CUDA device to use (export CUDA_VISIBLE_DEVICES=0,1,etc might be needed before)", default="7")
1

找到了，修改default值就可以，我这里是8卡4090，指定最后一个卡。