alexnasa commited on
Commit
2a9f7fd
·
verified ·
1 Parent(s): ea2f235

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -49,7 +49,8 @@ print("installing cuda toolkit")
49
  install_cuda_toolkit()
50
  print("finished")
51
 
52
- print(os.environ["CUDA_HOME"])
 
53
 
54
  # my_env = os.environ.copy()
55
  subprocess.run(["pip", "install","diso"], check=True)
 
49
  install_cuda_toolkit()
50
  print("finished")
51
 
52
+ header_path = "/usr/local/cuda/include/cuda_runtime.h"
53
+ print(f"{header_path} exists:", os.path.exists(header_path))
54
 
55
  # my_env = os.environ.copy()
56
  subprocess.run(["pip", "install","diso"], check=True)