Solved by SANTOSH_S in post #2 
              
              
                Hi, 
@Ratan you  can use nvidia-smi command to find out which GPU 0 or 1 has more memory. Accordingly you can follow one of the following ways to load the model to the manually to particular GPU: 
Inside the python file you can add: 
os.environ['CUDA_VISIBLE_DEVICES'] = "0" 
In command line 
CUDA_VI…