Fix python3 installation (#270)

AmpereComputingAI · Nov 26, 2024 · 9f898ca · 9f898ca
1 parent c5bb1a2
commit 9f898ca
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/setup_deb.sh b/setup_deb.sh
@@ -46,11 +46,17 @@ fi
 log "Installing system dependencies ..."
 sleep 1
 apt-get update -y
-apt-get install -y python3 python3-pip build-essential ffmpeg libsm6 libxext6 wget git unzip numactl libhdf5-dev cmake
+apt-get install -y build-essential ffmpeg libsm6 libxext6 wget git unzip numactl libhdf5-dev cmake
+if ! python3 -c ""; then
+    apt-get install -y python3 python3-pip
+fi
+if ! pip3 --version; then
+    apt-get install -y python3-pip
+fi
 PYTHON_VERSION=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[0:2])))')
 PYTHON_DEV_SEARCH=$(apt-cache search --names-only "python${PYTHON_VERSION}-dev")
 if [[ -n "$PYTHON_DEV_SEARCH" ]]; then
-    apt-get -y install "python${PYTHON_VERSION}-dev"
+    apt-get install -y "python${PYTHON_VERSION}-dev"
 fi
 log "done.\n"
 

diff --git a/setup_rhel.sh b/setup_rhel.sh
@@ -47,7 +47,7 @@ log "Installing system dependencies ..."
 sleep 1
 yum install epel-release || :
 yum groupinstall -y 'Development Tools'
-yum install -y python3 python3-devel python3-pip libSM libXext wget git unzip numactl hdf5-devel cmake gcc-c++ 
+yum install -y python3-devel python3-pip libSM libXext wget git unzip numactl hdf5-devel cmake gcc-c++ 
 git clone -b n4.3.7 https://github.com/FFmpeg/FFmpeg.git && cd FFmpeg && ./configure && make -j && make install && cd .. && rm -r FFmpeg
 log "done.\n"