jingyangcarl commited on
Commit
a7f3455
·
1 Parent(s): 80936bb
Files changed (1) hide show
  1. run.sh +1 -0
run.sh CHANGED
@@ -7,4 +7,5 @@ conda activate gradio-user
7
  # Start app.py
8
  echo "Starting app.py..."
9
  export OMP_NUM_THREADS=4 # default is a wrong value: 7500m
 
10
  python app.py
 
7
  # Start app.py
8
  echo "Starting app.py..."
9
  export OMP_NUM_THREADS=4 # default is a wrong value: 7500m
10
+ python -c "import torch; x=torch.rand(1, device='cuda'); print(x, x.device.type)"
11
  python app.py