search

gitmylo / audio-webui

A webui for different audio related Neural Networks
MIT License
973 stars 90 forks source link

[QUESTION] Specify a GPU #168

Open navarisun1982 opened 9 months ago

navarisun1982 commented 9 months ago

How to specify a GPU?

Hello, I have 2 GPU in my workstation 0: Tesla p40 24GB 1: Quadro k4200 4GB

My main GPU is Tesla, every time i run audio-webui, it insists to run using Quadro, even through the Nvidia control panel I select to run it with tesla p40, is there any solution for this, here is the error message:

Found GPU1 Quadro K4200 which is of Cuda capability 3.0.
PyTorch no longer supports this GPU because it is too old.
The minimum Cuda capability supported by this library is 3.7.
Mikec78660 commented 8 months ago

You can set the GPU that is visible in the shell with:

export CUDA_VISIBLE_DEVICES=0

or just:

CUDA_VISIBLE_DEVICES=0 ./run.sh

if it still wants to try to use the Quadro try setting the GPU to 1. Seems like some things list the GPUs in different orders. CUDA vs nvidia-smi vs lspci.

navarisun1982 commented 8 months ago

@Mikec78660 thx for support

SybarisUK commented 6 months ago

I'm having a similar issue in that my laptop has onboard intel UHD Graphics listed as 0, and my Nvidia card is 1. Audio-webui is just using the intel onboard GPU. I read the comments above, but not sure where you need to type these commands to change it. Can you give me a more step by step guide to this please?