Unanswered
Hi Everyone,
I Am Trying To Use Docker Mode For Trains-Agent, But It Seems That It Has Problem With The Use Of Multiple Gpus
This Is My Trains-Agent Command: Trains-Agent Daemon --Gpus 0,1 --Queue Dual_Gpu --Docker --Foreground
And It Gets The Error:
Doc
You can try running the trains-agent
with --gpus all
if you have two gpus on your machine.
The flag --gpus all
is used to assign all available gpus to the docker container.
Did you get the same error message? What do you have in the error under ‘device=XXX’?
What is the OS / docker / Nvidia drivers you have on the machine?
165 Views
0
Answers
4 years ago
one year ago