deezer / deezer/spleeter

[Discussion] How to set GPU device

Open
#425 2 comments 0 reactions 0 assignees View on GitHub
question
Dominant language
Python
Stars
28.4k
Forks
3.1k
PR merge metrics
No merged PRs in 30d

Description

I have multiple GPU's but I can't figure out how to set which GPU device separator should use.

In my python script I've tried adding `os.environ['CUDA_VISIBLE_DEVICES'] = "1"` but it does nothing.

I've also tried adding `device_count={'GPU': 1}` to `ConfigProto`, but it does nothing either.

Has anyone got a clue how to set which device to use?

I'm on Windows 10 with RTX2080

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.