Spaces:
Running
Running
File size: 480 Bytes
23dfebd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
#!/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
|