Cant get SLEAP to run on GPU1 instead of integrated graphics on Laptop

[HarryA-11]

I have a dedicated GPU1 on a Dell Precision 7550 (NVIDIA Quadro RTX 4000) and yet SLEAP still runs on GPU 0

python -c "import sleap; sleap.system_summary()"

GPUs: 1/1 available
Device: /physical_device:GPU:0
Available: True
Initalized: False
Memory growth: None

python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

Please can you help getting some use from this dedicated GPU. I am limited by the amount I can do as a non admin user on a company issued laptop but if this is done via bios settings I am sure they will help. Thank you

[github-actions[bot]]

Thank you for starting a new discussion!

We appreciate your input and will review it soon.

Warning

A friendly reminder that this is a public forum. Please be cautious when clicking links, downloading files, or running scripts posted by others.

• Always verify the credibility of links and code.
• Avoid running scripts or installing files from untrusted sources.
• If you're unsure, ask for clarification before proceeding.

Stay safe and happy SLEAPing!

Best regards,
The Team

[gitttt-1234]

Hi @HarryA-11,

Tensorflow doesn't support accelerated operations on integrated GPUs and thus won't recognize them. Tensorflow has native support for NVIDIA GPUs (with CUDA). You can verify this by running the below command:
nvidia-smi
The GPU shown in the output is the one detected/ used by tensorflow (and thus, by SLEAP).
Let us know if you have any questions!

Thanks,

Divya