Browse Source

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

fix GPU numbering
Jason Antic 6 years ago
parent
commit
c1f59817b9
3 changed files with 3 additions and 3 deletions
  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' "
    ]
   },
   {