run.sh

#!/bin/bash

CONDA_ENV=$(head -1 /code/environment.yml | cut -d" " -f2)
eval "$(conda shell.bash hook)"
conda activate gradio-user
export OMP_NUM_THREADS=4 # default is a wrong value: 7500m

conda install -n gradio-user pytorch3d=0.7.7 -c pytorch3d -c conda-forge
conda install -n gradio-user -c conda-forge open-clip-torch pytorch-lightning

# Start app.py
echo "Starting app.py..."
python -c "import torch; x=torch.rand(1, device='cuda'); print(x, x.device.type)"
python app.py