abreza commited on
Commit
738777e
1 Parent(s): fdd8235
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ model_option = gr.Radio(options, value="dino16",
89
 
90
  def find_cuda_home():
91
  try:
92
- output = subprocess.check_output(['ls', '/usr/local/']).decode()
93
  print(output)
94
  for line in output.split('\n'):
95
  if 'Cuda compilation tools' in line:
 
89
 
90
  def find_cuda_home():
91
  try:
92
+ output = subprocess.check_output(['which', 'nvcc']).decode()
93
  print(output)
94
  for line in output.split('\n'):
95
  if 'Cuda compilation tools' in line: