Merge pull request Linaqruf#179 from huytd2k/patch-1

fix LD_LIBRARY_PATH raise an error
Azjou · Apr 6, 2023 · aad097d · aad097d
2 parents 138a028 + 3376623
commit aad097d
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/kohya-LoRA-dreambooth.ipynb b/kohya-LoRA-dreambooth.ipynb
@@ -209,7 +209,7 @@
         "    os.environ[\"LD_PRELOAD\"] = \"libtcmalloc.so\"\n",
         "    os.environ[\"TF_CPP_MIN_LOG_LEVEL\"] = \"3\"\n",
         "    os.environ[\"BITSANDBYTES_NOWELCOME\"] = \"1\"  \n",
-        "    os.environ[\"LD_LIBRARY_PATH\"] = \"/usr/local/cuda/lib64/:$LD_LIBRARY_PATH\"\n",
+        "    os.environ[\"LD_LIBRARY_PATH\"] = \"/usr/local/cuda/lib64/:\" + os.environ[\"LD_LIBRARY_PATH\"]\n",
         "    os.environ[\"SAFETENSORS_FAST_GPU\"] = \"1\"\n",
         "\n",
         "main()\n"

diff --git a/kohya-LoRA-finetuner.ipynb b/kohya-LoRA-finetuner.ipynb
@@ -209,7 +209,7 @@
         "    os.environ[\"LD_PRELOAD\"] = \"libtcmalloc.so\"\n",
         "    os.environ[\"TF_CPP_MIN_LOG_LEVEL\"] = \"3\"\n",
         "    os.environ[\"BITSANDBYTES_NOWELCOME\"] = \"1\"  \n",
-        "    os.environ[\"LD_LIBRARY_PATH\"] = \"/usr/local/cuda/lib64/:$LD_LIBRARY_PATH\"\n",
+        "    os.environ[\"LD_LIBRARY_PATH\"] = \"/usr/local/cuda/lib64/:\" + os.environ[\"LD_LIBRARY_PATH\"]\n",
         "    os.environ[\"SAFETENSORS_FAST_GPU\"] = \"1\"\n",
         "\n",
         "main()\n"

diff --git a/kohya-dreambooth.ipynb b/kohya-dreambooth.ipynb
@@ -209,7 +209,7 @@
         "    os.environ[\"LD_PRELOAD\"] = \"libtcmalloc.so\"\n",
         "    os.environ[\"TF_CPP_MIN_LOG_LEVEL\"] = \"3\"\n",
         "    os.environ[\"BITSANDBYTES_NOWELCOME\"] = \"1\"  \n",
-        "    os.environ[\"LD_LIBRARY_PATH\"] = \"/usr/local/cuda/lib64/:$LD_LIBRARY_PATH\"\n",
+        "    os.environ[\"LD_LIBRARY_PATH\"] = \"/usr/local/cuda/lib64/:\" + os.environ[\"LD_LIBRARY_PATH\"]\n",
         "    os.environ[\"SAFETENSORS_FAST_GPU\"] = \"1\"\n",
         "\n",
         "main()\n"

diff --git a/kohya-trainer.ipynb b/kohya-trainer.ipynb
@@ -209,7 +209,7 @@
         "    os.environ[\"LD_PRELOAD\"] = \"libtcmalloc.so\"\n",
         "    os.environ[\"TF_CPP_MIN_LOG_LEVEL\"] = \"3\"\n",
         "    os.environ[\"BITSANDBYTES_NOWELCOME\"] = \"1\"  \n",
-        "    os.environ[\"LD_LIBRARY_PATH\"] = \"/usr/local/cuda/lib64/:$LD_LIBRARY_PATH\"\n",
+        "    os.environ[\"LD_LIBRARY_PATH\"] = \"/usr/local/cuda/lib64/:\" + os.environ[\"LD_LIBRARY_PATH\"]\n",
         "    os.environ[\"SAFETENSORS_FAST_GPU\"] = \"1\"\n",
         "\n",
         "main()\n"