fix `pip: not found`

app.py

@@ -35,6 +35,7 @@ from download_checkpoints import main as download_checkpoints
 from test_environment import main as check_environment
 from test_environment import setup_environment
 
+setup_environment()
 
 # setup env
 os.environ["CUDA_HOME"] = "/usr/local/cuda"
@@ -42,9 +43,7 @@ os.environ["LD_LIBRARY_PATH"] = "$CUDA_HOME/lib:$CUDA_HOME/lib64:$LD_LIBRARY_PAT
 os.environ["PATH"] = "$CUDA_HOME/bin:/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:$PATH"
 
 if not check_environment():
-    setup_environment()
-    if not check_environment():
-        sys.exit(1)
+    sys.exit(1)
 
 os.environ["TOKENIZERS_PARALLELISM"] = "false"  # Workaround to suppress MP warning
 

test_environment.py

@@ -18,21 +18,23 @@ import importlib
 import os
 import sys
 
+PIP = f"{sys.executable} -m pip"
+print(PIP)
 
 def setup_environment():
     os.system("apt-get update && apt-get install -qqy libmagickwand-dev")
 
     # install packages
     # os.system(
-    #     'export FLASH_ATTENTION_SKIP_CUDA_BUILD=FALSE && \
-    #     pip install --timeout=1000000000 --no-build-isolation "flash-attn<=2.7.4.post1"'
+    #     f'export FLASH_ATTENTION_SKIP_CUDA_BUILD=FALSE && \
+    #     {PIP} install --timeout=1000000000 --no-build-isolation "flash-attn<=2.7.4.post1"'
     # )
     os.system(
-        "pip install --timeout=1000000000 \
+        f"{PIP} install --timeout=1000000000 \
         https://download.pytorch.org/whl/cu128/flashinfer/flashinfer_python-0.2.5%2Bcu128torch2.7-cp38-abi3-linux_x86_64.whl"
     )
-    os.system('export VLLM_ATTENTION_BACKEND=FLASHINFER && pip install "vllm==0.9.0"')
-    os.system('pip install "decord==0.6.0"')
+    os.system(f'export VLLM_ATTENTION_BACKEND=FLASHINFER && {PIP} install "vllm==0.9.0"')
+    os.system(f'{PIP} install "decord==0.6.0"')
 
     os.system(
         "export CONDA_PREFIX=/usr/local/cuda && \
@@ -43,11 +45,11 @@ def setup_environment():
         ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/python3.10"
     )
 
-    os.system('pip install --timeout=1000000000 --no-build-isolation "transformer-engine[pytorch]"')
-    os.system('pip install --timeout=1000000000 "decord==0.6.0"')
+    os.system(f'{PIP} install --timeout=1000000000 --no-build-isolation "transformer-engine[pytorch]"')
+    os.system(f'{PIP} install --timeout=1000000000 "decord==0.6.0"')
 
     # os.system(
-    #     'pip install --timeout=1000000000 \
+    #     f'{PIP} install --timeout=1000000000 \
     #     "git+https://github.com/nvidia-cosmos/cosmos-transfer1@e4055e39ee9c53165e85275bdab84ed20909714a"'
     # )