瀏覽代碼

fix GPU numbering

jqueguiner 6 年之前
父節點
當前提交
1f2a5eff5f
共有 3 個文件被更改,包括 3 次插入3 次删除
  1. 1 1
      ColorizeTrainingVideo.ipynb
  2. 1 1
      ImageColorizer.ipynb
  3. 1 1
      ImageColorizerArtisticTests.ipynb

+ 1 - 1
ColorizeTrainingVideo.ipynb

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

+ 1 - 1
ImageColorizer.ipynb

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

+ 1 - 1
ImageColorizerArtisticTests.ipynb

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