Update atsetup.sh

This commit is contained in:
erew123 authored and GitHub committed 2024-11-26 00:23:42 +00:00
1 parent 7c23b043d9
commit 393bff6ab9
1 file changed
+1 -36
+1 -36
View File
@@ -345,7 +345,7 @@ install_custom_standalone() {
conda install -y nvidia/label/cuda-12.1.0::cuda-toolkit=12.1
conda install -y pytorch::faiss-cpu
conda install -y -c conda-forge "ffmpeg=*=*gpl*"
conda install -y conda-forge::ffmpeg
conda install -y -c conda-forge "ffmpeg=*=h*_*" --no-deps
echo
echo
echo
@@ -354,41 +354,6 @@ install_custom_standalone() {
# Define the environment path
env_path="${INSTALL_ENV_DIR}/lib"
# List of broken symlinks and their correct targets
declare -A symlinks=(
["libcufile.so"]="libcufile.so.1.10.0"
["libcufile_rdma.so"]="libcufile_rdma.so.1.10.0"
["libcudart.so"]="libcudart.so.12"
["libnvjpeg.so"]="libnvjpeg.so.12.1.1.14"
["libcurand.so"]="libcurand.so.10.3.6.39"
["libnvJitLink.so"]="libnvJitLink.so.12.1.105"
["libnvrtc-builtins.so"]="libnvrtc-builtins.so.12.1.105"
["libnvrtc.so"]="libnvrtc.so.12.1.105"
)
# Function to fix broken symlinks
fix_broken_symlinks() {
echo "Fixing broken symlinks..."
for link in "${!symlinks[@]}"; do
target="${symlinks[$link]}"
if [ -L "$env_path/$link" ] && [ ! -e "$env_path/$link" ]; then
echo "Removing broken link: $env_path/$link"
rm "$env_path/$link"
fi
if [ -e "$env_path/$target" ]; then
echo "Creating new symlink: $env_path/$link -> $env_path/$target"
ln -s "$env_path/$target" "$env_path/$link"
else
echo "Target file does not exist: $env_path/$target"
fi
done
echo "Verification of symbolic links:"
ls -l $env_path | grep -E "libcufile.so|libcudart.so|libcufile_rdma.so|libnvjpeg.so|libcurand.so|libnvJitLink.so|libnvrtc-builtins.so|libnvrtc.so"
}
# Call the function to fix broken symlinks
fix_broken_symlinks
echo " Installing additional requirements."
echo
pip install -r system/requirements/requirements_standalone.txt