fix CUDA_VISIBLE_DEVICE=-1 bug

shijun18 · Oct 27, 2020 · acdfa96 · acdfa96
1 parent c67e05f
commit acdfa96
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dygraph/paddleseg/utils/env/sys_env.py b/dygraph/paddleseg/utils/env/sys_env.py
@@ -103,6 +103,8 @@ def get_sys_env():
 
         env_info['CUDA_VISIBLE_DEVICES'] = os.environ.get(
             'CUDA_VISIBLE_DEVICES')
+        if gpu_nums == 0:
+            os.environ['CUDA_VISIBLE_DEVICES'] = ''
         env_info['GPU'] = _get_gpu_info()
 
     try: