setup_envs.sh

#!/bin/bash

# Script to verify PyTorch dependencies in Docker container
# This script checks existing PyTorch/numpy versions and modifies requirements.txt
# to avoid reinstalling them, using the base image versions instead.

set -e

echo "=== Starting dependency verification script ==="

# Check existing PyTorch and numpy
echo "=== Checking existing PyTorch and numpy ==="
python3 -c "import torch; import torchvision; import numpy; print(f'Existing torch: {torch.__version__}, torchvision: {torchvision.__version__}, numpy: {numpy.__version__}')" || echo "check failed"

# Verify requirements.txt exists
if [ ! -f "requirements.txt" ]; then
    echo "ERROR: requirements.txt not found in current directory"
    exit 1
fi

echo "=== Original requirements.txt ==="
cat requirements.txt

# Modify requirements to remove torch, torchvision, and numpy (to use system versions)
echo "=== Modifying requirements.txt to remove torch, torchvision, and numpy ==="
sed -i '/^torch==/d' requirements.txt
sed -i '/^torch>=/d' requirements.txt
sed -i '/^torch$/d' requirements.txt
sed -i '/^torchvision/d' requirements.txt
sed -i '/^numpy/d' requirements.txt

echo "=== Modified requirements.txt ==="
cat requirements.txt

# Install modified requirements
echo "=== Installing modified requirements ==="
pip install --no-cache-dir --ignore-installed blinker -r requirements.txt

echo "=== Dependency verification completed successfully ==="