jingyangcarl commited on
Commit
2fcceef
·
1 Parent(s): b17cab7
Files changed (1) hide show
  1. run.sh +1 -1
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