fix GPU numbering
@@ -23,7 +23,7 @@
"outputs": [],
"source": [
"import os\n",
- "os.environ['CUDA_VISIBLE_DEVICES']='2' "
+ "os.environ['CUDA_VISIBLE_DEVICES']='0' "
]
},
{
@@ -7,7 +7,7 @@
- "os.environ['CUDA_VISIBLE_DEVICES']='1' "