From f83d5a41da545868a351decf58bf82d34715207b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pablo=20Rodr=C3=ADguez=20Flores?= Date: Thu, 23 May 2024 11:11:09 +0100 Subject: [PATCH] Fixed minor path error --- packaging/rpm/rb-aioutliers.spec | 4 ++-- resources/src/setup_tensorflow_and_cuda.sh | 3 +-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/packaging/rpm/rb-aioutliers.spec b/packaging/rpm/rb-aioutliers.spec index ea2f19a..f9a727b 100644 --- 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 index cba44ea..573ac87 100755 --- 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