Spaces:
Sleeping
Sleeping
File size: 328 Bytes
1d4bf8c a7f3455 1d4bf8c | 1 2 3 4 5 6 7 8 9 10 11 12 | #!/bin/bash
CONDA_ENV=$(head -1 /code/environment.yml | cut -d" " -f2)
eval "$(conda shell.bash hook)"
conda activate gradio-user
# Start app.py
echo "Starting app.py..."
export OMP_NUM_THREADS=4 # default is a wrong value: 7500m
python -c "import torch; x=torch.rand(1, device='cuda'); print(x, x.device.type)"
python app.py
|