Add CUDA_VISIBLE_DEVICES environment variable when using the --gpus flag#345

Merged

sergey-serebryakov merged 8 commits intomlcommons:masterfrom davidjurado:feature/docker_gpu

Jan 5, 2024

+21-4

Commits on Nov 15, 2023

Add CUDA_VISIBLE_DEVICES enviroment variable when using the --gpus flag
davidjurado
committed

Commits on Nov 28, 2023

Fix logic when defining enviroment variable
davidjurado
committed

Commits on Nov 30, 2023

Fix validation when using --gpus flag
davidjurado
committed

Commits on Dec 8, 2023

Commits on Dec 12, 2023

Commits on Dec 19, 2023

Fix parser logic
davidjurado
committed