Solved by mrshenli in post #4 
              
              
                Have you tried setting CUDA_VISIBLE_DEVICES env var before launching the process? It would be more clear if you share some minimum code snippet