ford442 commited on
Commit
549ff6f
·
verified ·
1 Parent(s): 45f8913

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -20,16 +20,18 @@ os.environ["PATH"] = conda_bin + os.pathsep + os.environ["PATH"]
20
  # Activate the base environment (adjust if needed)
21
  os.system(f'{conda_bin}/conda init --all')
22
  os.system(f'{conda_bin}/conda activate base')
 
23
  os.system(f'{conda_bin}/conda install nvidia/label/cuda-12.4.0::cuda-libraries')
24
  os.system(f'{conda_bin}/conda install nvidia/label/cuda-12.4.0::cuda-libraries-dev')
25
  os.system(f'{conda_bin}/conda install nvidia/label/cuda-12.4.0::cuda-cudart')
26
  os.system(f'{conda_bin}/conda install nvidia/label/cuda-12.4.0::cuda-cudart-dev')
27
  os.system(f'{conda_bin}/conda install nvidia/label/cuda-12.4.0::cuda-nvcc')
28
-
29
- subprocess.run(['pip', 'install', 'git+https://github.com/ford442/hidet.git@thread'])
30
 
31
  #os.system(f'{conda_bin}/conda install pytorch::pytorch-cuda')
32
  #os.system(f'{conda_bin}/conda install rcdr_py37::tensorrt')
 
33
 
34
  import random
35
  import uuid
 
20
  # Activate the base environment (adjust if needed)
21
  os.system(f'{conda_bin}/conda init --all')
22
  os.system(f'{conda_bin}/conda activate base')
23
+ os.system(f'{conda_bin}/conda install nvidia/label/cudnn-9.3.0::cudnn')
24
  os.system(f'{conda_bin}/conda install nvidia/label/cuda-12.4.0::cuda-libraries')
25
  os.system(f'{conda_bin}/conda install nvidia/label/cuda-12.4.0::cuda-libraries-dev')
26
  os.system(f'{conda_bin}/conda install nvidia/label/cuda-12.4.0::cuda-cudart')
27
  os.system(f'{conda_bin}/conda install nvidia/label/cuda-12.4.0::cuda-cudart-dev')
28
  os.system(f'{conda_bin}/conda install nvidia/label/cuda-12.4.0::cuda-nvcc')
29
+ #subprocess.run(['pip', 'install', 'git+https://github.com/hidet-org/hidet.git'])
30
+ #subprocess.run(['pip', 'install', 'git+https://github.com/ford442/hidet.git@thread'])
31
 
32
  #os.system(f'{conda_bin}/conda install pytorch::pytorch-cuda')
33
  #os.system(f'{conda_bin}/conda install rcdr_py37::tensorrt')
34
+ subprocess.run(['sh', './hidet.sh'])
35
 
36
  import random
37
  import uuid