From b3e8e9537fef55ea491c74740ab4e0d1f57815b6 Mon Sep 17 00:00:00 2001
From: bmaltais <bernard@ducourier.com>
Date: Sat, 24 Jun 2023 19:51:40 -0400
Subject: [PATCH] Updates for runpod

---
 setup.sh | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/setup.sh b/setup.sh
index 5c67a82..cb34d32 100755
--- a/setup.sh
+++ b/setup.sh
@@ -333,7 +333,7 @@ configure_accelerate() {
     else
       echo "Could not place the accelerate configuration file. Please configure manually."
       sleep 2
-      accelerate config
+      bash accelerate config
     fi
   fi
 }
@@ -558,21 +558,21 @@ if [[ "$OSTYPE" == "lin"* ]]; then
 
   # We need just a little bit more setup for non-interactive environments
   if [ "$RUNPOD" = true ]; then
-    if inDocker; then
-      # We get the site-packages from python itself, then cut the string, so no other code changes required.
-      VENV_DIR=$(python -c "import site; print(site.getsitepackages()[0])")
-      VENV_DIR="${VENV_DIR%/lib/python3.10/site-packages}"
-    fi
+    # if inDocker; then
+    #   # We get the site-packages from python itself, then cut the string, so no other code changes required.
+    #   VENV_DIR=$(python -c "import site; print(site.getsitepackages()[0])")
+    #   VENV_DIR="${VENV_DIR%/lib/python3.10/site-packages}"
+    # fi
 
-    # Symlink paths
-    libnvinfer_plugin_symlink="$VENV_DIR/lib/python3.10/site-packages/tensorrt/libnvinfer_plugin.so.7"
-    libnvinfer_symlink="$VENV_DIR/lib/python3.10/site-packages/tensorrt/libnvinfer.so.7"
-    libcudart_symlink="$VENV_DIR/lib/python3.10/site-packages/nvidia/cuda_runtime/lib/libcudart.so.11.0"
+    # # Symlink paths
+    # libnvinfer_plugin_symlink="$VENV_DIR/lib/python3.10/site-packages/tensorrt/libnvinfer_plugin.so.7"
+    # libnvinfer_symlink="$VENV_DIR/lib/python3.10/site-packages/tensorrt/libnvinfer.so.7"
+    # libcudart_symlink="$VENV_DIR/lib/python3.10/site-packages/nvidia/cuda_runtime/lib/libcudart.so.11.0"
 
-    #Target file paths
-    libnvinfer_plugin_target="$VENV_DIR/lib/python3.10/site-packages/tensorrt/libnvinfer_plugin.so.8"
-    libnvinfer_target="$VENV_DIR/lib/python3.10/site-packages/tensorrt/libnvinfer.so.8"
-    libcudart_target="$VENV_DIR/lib/python3.10/site-packages/nvidia/cuda_runtime/lib/libcudart.so.12"
+    # #Target file paths
+    # libnvinfer_plugin_target="$VENV_DIR/lib/python3.10/site-packages/tensorrt/libnvinfer_plugin.so.8"
+    # libnvinfer_target="$VENV_DIR/lib/python3.10/site-packages/tensorrt/libnvinfer.so.8"
+    # libcudart_target="$VENV_DIR/lib/python3.10/site-packages/nvidia/cuda_runtime/lib/libcudart.so.12"
 
     # echo "Checking symlinks now."
     # create_symlinks "$libnvinfer_plugin_symlink" "$libnvinfer_plugin_target"