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