Преглед на файлове

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' "
    ]
   },
   {