rodrigomasini commited on
Commit
ce75bc0
1 Parent(s): b2e5e8f

Update app_v3.py

Browse files
Files changed (1) hide show
  1. app_v3.py +2 -1
app_v3.py CHANGED
@@ -7,7 +7,8 @@ import torch
7
  import subprocess
8
 
9
  os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:128'
10
- os.environ['CUDA_VISIBLE_DEVICES'] = '0' # Assuming you want to use GPU 0
 
11
 
12
 
13
  # Define pretrained and quantized model directories
 
7
  import subprocess
8
 
9
  os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:128'
10
+ os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
11
+ os.environ["CUDA_VISIBLE_DEVICES"]="1"
12
 
13
 
14
  # Define pretrained and quantized model directories