Spaces:
Build error
Build error
Commit
·
2fcceef
1
Parent(s):
b17cab7
update
Browse files
run.sh
CHANGED
@@ -3,12 +3,12 @@
|
|
3 |
CONDA_ENV=$(head -1 /code/environment.yml | cut -d" " -f2)
|
4 |
eval "$(conda shell.bash hook)"
|
5 |
conda activate gradio-user
|
|
|
6 |
|
7 |
conda install -n gradio-user pytorch3d=0.7.7 -c pytorch3d -c conda-forge
|
8 |
conda install -n gradio-user -c conda-forge open-clip-torch pytorch-lightning
|
9 |
|
10 |
# Start app.py
|
11 |
echo "Starting app.py..."
|
12 |
-
export OMP_NUM_THREADS=4 # default is a wrong value: 7500m
|
13 |
python -c "import torch; x=torch.rand(1, device='cuda'); print(x, x.device.type)"
|
14 |
python app.py
|
|
|
3 |
CONDA_ENV=$(head -1 /code/environment.yml | cut -d" " -f2)
|
4 |
eval "$(conda shell.bash hook)"
|
5 |
conda activate gradio-user
|
6 |
+
export OMP_NUM_THREADS=4 # default is a wrong value: 7500m
|
7 |
|
8 |
conda install -n gradio-user pytorch3d=0.7.7 -c pytorch3d -c conda-forge
|
9 |
conda install -n gradio-user -c conda-forge open-clip-torch pytorch-lightning
|
10 |
|
11 |
# Start app.py
|
12 |
echo "Starting app.py..."
|
|
|
13 |
python -c "import torch; x=torch.rand(1, device='cuda'); print(x, x.device.type)"
|
14 |
python app.py
|