Fixed minor path error

redBorder · May 23, 2024 · f83d5a4 · f83d5a4
1 parent 5734aef
commit f83d5a4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/packaging/rpm/rb-aioutliers.spec b/packaging/rpm/rb-aioutliers.spec
@@ -69,10 +69,10 @@ source /opt/rb-aioutliers/aioutliers/bin/activate
 pip3 install -r /opt/rb-aioutliers/resources/src/requirements.txt
 
 # Add NVIDIA libraries to LD_LIBRARY_PATH (Necessary for using TensorFlow with GPU)
-cp /opt/rb-aioutliers/resources/src/setup_tensorflow_and_cuda.sh /opt/rb-aioutliers/bin/setup_tensorflow_and_cuda.sh
+cp /opt/rb-aioutliers/resources/src/setup_tensorflow_and_cuda.sh /opt/rb-aioutliers/aioutliers/bin/setup_tensorflow_and_cuda.sh
 
 # Append the sourcing of setup script to activate script
-echo "source /opt/rb-aioutliers/bin/setup_tensorflow_and_cuda.sh" | tee -a /opt/rb-aioutliers/aioutliers/bin/activate
+echo "source /opt/rb-aioutliers/aioutliers/bin/setup_tensorflow_and_cuda.sh" | tee -a /opt/rb-aioutliers/aioutliers/bin/activate
 
 # Deactivate and reactivate to apply changes
 deactivate

diff --git a/resources/src/setup_tensorflow_and_cuda.sh b/resources/src/setup_tensorflow_and_cuda.sh
@@ -1,8 +1,7 @@
 #!/bin/bash
 
 # Set the base directory for NVIDIA packages within the virtual environment using wildcard for Python version
-NVIDIA_PACKAGE_DIR="$(echo "$VIRTUAL_ENV")/lib/python*/site-packages/nvidia"
-echo $NVIDIA_PACKAGE_DIR
+NVIDIA_PACKAGE_DIR="$VIRTUAL_ENV/lib/python*/site-packages/nvidia"
 # Export LD_LIBRARY_PATH for each lib directory found in the NVIDIA_PACKAGE_DIR
 for dir in $NVIDIA_PACKAGE_DIR/*; do
     if [ -d "$dir/lib" ]; then