How to select a specific single GPU in a multi-GPU system

[rlewisfr]

Thought I would drop this here because I was looking and could not find.

I have a 3060ti (8gb) and a 3060 (12gb). when I first started training, it sent my 3060ti into helicopter take-off mode, since it is my system primary.

To select the alternate GPU, I borrowed my code from Koya_ss launch. Edit your OneTrainer\start-ui.bat in the following manner:

:activate_venv
set CUDA_VISIBLE_DEVICES=1
echo activating venv %VENV_DIR%
set PYTHON="%VENV_DIR%\Scripts\python.exe"

Add the CUDA_VISIBLE_DEVICES line to the batch file. System default is 0, the alternate is 1

This will probably apply to a very small portion of people, but thought I would help those in the same boat as I.

Cheers!

[Tchernobill]

In the General tab, at Train Device, I was able to select which card to use by changing "cuda" for "cuda:0" or "cuda:1"

[randomtask2000]

are you able to use multiple gpu's like you can with Kohya_ss? Will it take cuda:0,1?

[mx]

are you able to use multiple gpu's like you can with Kohya_ss? Will it take cuda:0,1?

Not at this time.

[riffmaster-2001]

This worked for me:
in start-ui.sh I put this at the top

export TORCH_USE_CUDA_DSA=1
export CUDA_VISIBLE_DEVICES=1

and in my train device I just kept "cuda". When I started it back up I was able to get it to use my other GPU Card