浏览代码

Merge pull request #98 from jqueguiner/fix-GPU-numbering

fix GPU numbering
Jason Antic 6 年之前
父节点
当前提交
c1f59817b9
共有 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' "
    ]
   },
   {