abreza commited on
Commit
fdd8235
1 Parent(s): 788fe60
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(['nvcc', '--version']).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(['ls', '/usr/local/']).decode()
93
  print(output)
94
  for line in output.split('\n'):
95
  if 'Cuda compilation tools' in line: