Train model by using a specific GPU

3

Hi,

Setting the env variable within python won’t work, it needs to be set in the shell before you run your script.
Also, this might help as well.