Update python/dask_cudf/dask_cudf/io/parquet.py

rapidsai · Nov 19, 2024 · fa03396 · fa03396
1 parent 0ecf6dd
commit fa03396
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dask_cudf/dask_cudf/io/parquet.py b/python/dask_cudf/dask_cudf/io/parquet.py
@@ -47,7 +47,7 @@ def _get_device_size():
 
         pynvml.nvmlInit()
         index = os.environ.get("CUDA_VISIBLE_DEVICES", "0").split(",")[0]
-        if not index.isnumeric():
+        if index and not index.isnumeric():
             # This means index is UUID. This works for both MIG and non-MIG device UUIDs.
             handle = pynvml.nvmlDeviceGetHandleByUUID(str.encode(index))
         else: