diff --git a/fast-kohya-trainer.ipynb b/fast-kohya-trainer.ipynb
index 96be93a2..cecf0d3b 100644
--- a/fast-kohya-trainer.ipynb
+++ b/fast-kohya-trainer.ipynb
@@ -820,7 +820,7 @@
     "  !pip -q install --upgrade -r requirements.txt\n",
     "\n",
     "  if install_xformers:\n",
-    "    !pip install xformers==0.0.16 triton==2.0.0\n",
+    "    !pip install xformers==0.0.18 triton\n",
     "\n",
     "  from accelerate.utils import write_basic_config\n",
     "  if not os.path.exists(accelerate_config):\n",
diff --git a/kohya-LoRA-dreambooth.ipynb b/kohya-LoRA-dreambooth.ipynb
index 9eda87b9..261654b8 100644
--- a/kohya-LoRA-dreambooth.ipynb
+++ b/kohya-LoRA-dreambooth.ipynb
@@ -209,9 +209,12 @@
         "    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/:\" + os.environ[\"LD_LIBRARY_PATH\"]\n",
         "    os.environ[\"SAFETENSORS_FAST_GPU\"] = \"1\"\n",
         "\n",
+        "    cuda_path = \"/usr/local/cuda-11.8/targets/x86_64-linux/lib/\"\n",
+        "    ld_library_path = os.environ.get(\"LD_LIBRARY_PATH\", \"\")\n",
+        "    os.environ[\"LD_LIBRARY_PATH\"] = f\"{ld_library_path}:{cuda_path}\"\n",
+        "\n",
         "main()\n"
       ]
     },
diff --git a/kohya-LoRA-finetuner.ipynb b/kohya-LoRA-finetuner.ipynb
index 95e35c9d..9cc58d5c 100644
--- a/kohya-LoRA-finetuner.ipynb
+++ b/kohya-LoRA-finetuner.ipynb
@@ -209,9 +209,12 @@
         "    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/:\" + os.environ[\"LD_LIBRARY_PATH\"]\n",
         "    os.environ[\"SAFETENSORS_FAST_GPU\"] = \"1\"\n",
         "\n",
+        "    cuda_path = \"/usr/local/cuda-11.8/targets/x86_64-linux/lib/\"\n",
+        "    ld_library_path = os.environ.get(\"LD_LIBRARY_PATH\", \"\")\n",
+        "    os.environ[\"LD_LIBRARY_PATH\"] = f\"{ld_library_path}:{cuda_path}\"\n",
+        "\n",
         "main()\n"
       ]
     },
diff --git a/kohya-dreambooth.ipynb b/kohya-dreambooth.ipynb
index 1d68de82..d3bfb385 100644
--- a/kohya-dreambooth.ipynb
+++ b/kohya-dreambooth.ipynb
@@ -209,9 +209,12 @@
         "    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/:\" + os.environ[\"LD_LIBRARY_PATH\"]\n",
         "    os.environ[\"SAFETENSORS_FAST_GPU\"] = \"1\"\n",
         "\n",
+        "    cuda_path = \"/usr/local/cuda-11.8/targets/x86_64-linux/lib/\"\n",
+        "    ld_library_path = os.environ.get(\"LD_LIBRARY_PATH\", \"\")\n",
+        "    os.environ[\"LD_LIBRARY_PATH\"] = f\"{ld_library_path}:{cuda_path}\"\n",
+        "\n",
         "main()\n"
       ]
     },
diff --git a/kohya-trainer.ipynb b/kohya-trainer.ipynb
index 0292874a..c3afccf3 100644
--- a/kohya-trainer.ipynb
+++ b/kohya-trainer.ipynb
@@ -209,9 +209,13 @@
         "    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/:\" + os.environ[\"LD_LIBRARY_PATH\"]\n",
         "    os.environ[\"SAFETENSORS_FAST_GPU\"] = \"1\"\n",
         "\n",
+        "    cuda_path = \"/usr/local/cuda-11.8/targets/x86_64-linux/lib/\"\n",
+        "    ld_library_path = os.environ.get(\"LD_LIBRARY_PATH\", \"\")\n",
+        "    os.environ[\"LD_LIBRARY_PATH\"] = f\"{ld_library_path}:{cuda_path}\"\n",
+        "\n",
+        "\n",
         "main()\n"
       ]
     },