modify CUDA_VISIBLE_DEVICES

rapidsai · Nov 6, 2024 · 349014c · 349014c
1 parent a4b9f7e
commit 349014c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cugraph-dgl/examples/dataset_from_disk_cudf.ipynb b/python/cugraph-dgl/examples/dataset_from_disk_cudf.ipynb
@@ -18,7 +18,7 @@
    "outputs": [],
    "source": [
     "import os\n",
-    "os.environ[\"CUDA_VISIBLE_DEVICES\"]=\"4\"\n",
+    "os.environ[\"CUDA_VISIBLE_DEVICES\"]=\"0\"\n",
     "import rmm\n",
     "import torch\n",
     "from rmm.allocators.torch import rmm_torch_allocator\n",