zhiweili commited on
Commit
3047a5c
Β·
1 Parent(s): 9114cb3

show torch

Browse files
Files changed (1) hide show
  1. app_onediff.py +1 -1
app_onediff.py CHANGED
@@ -45,7 +45,7 @@ def runcmd(cmd, verbose = False, *args, **kwargs):
45
 
46
  print(os.environ.get('CUDA_PATH'))
47
  print(os.environ.get('CUDA_HOME'))
48
- os.system("pip show cuda")
49
 
50
  os.system("CUDA_HOME=/usr/local/cuda-12.1 python3 -m pip install stable-fast")
51
 
 
45
 
46
  print(os.environ.get('CUDA_PATH'))
47
  print(os.environ.get('CUDA_HOME'))
48
+ os.system("pip show torch")
49
 
50
  os.system("CUDA_HOME=/usr/local/cuda-12.1 python3 -m pip install stable-fast")
51