Hi, I don't know how to use --gpuName flag to use only one out of two gpu available my machine. Could anyone please kindly explain that to me.
--gpuName
Try this instead : CUDA_VISIBLE_DEVICES=1 python3 flow --train . . .
CUDA_VISIBLE_DEVICES=1 python3 flow --train . . .
Most helpful comment
Try this instead :
CUDA_VISIBLE_DEVICES=1 python3 flow --train . . .