setup.sh

#!/bin/bash

# Add Environment Variables
# Add PATH and LD_LIBRARY_PATH environment variables (For nvcc; CUDA 12.2)
echo "Setting up CUDA environment variables..."
export PATH=/usr/local/cuda-12.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:$LD_LIBRARY_PATH

# Update system package list
echo "Updating system packages..."
apt-get update

# Install system-wide dependencies
echo "Installing system-wide dependencies..."
# apt-get install -y package1 package2

# Install Miniconda
# wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh

# Check if pip is installed, and install it if it isn't
# if ! command -v pip &> /dev/null
# then
#     echo "pip could not be found, installing..."
#     sudo apt-get install -y python3-pip
# fi

# pip install virtualenv
# python3 -m virtualenv venv
# . venv/bin/activate

echo "Updating pip..."
python3 -m pip install --upgrade pip

# Install Python dependencies
echo "Installing Python dependencies from requirements.txt..."
pip install ninja

# Default requirements
pip install git+https://github.com/NVlabs/tiny-cuda-nn.git#subdirectory=bindings/torch
RUN pip install nerfacc -f https://nerfacc-bucket.s3.us-west-2.amazonaws.com/whl/torch-2.0.0_cu118.html

pip install -r requirements.txt

echo "Setup completed successfully!"