iofu728 commited on
Commit
8b57e3a
1 Parent(s): b215053

Feature(MInference): add nvcc env

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -4,6 +4,14 @@ subprocess.run(
4
  "export CPATH=$CPATH:/usr/local/cuda/include",
5
  shell=True,
6
  )
 
 
 
 
 
 
 
 
7
  subprocess.run(
8
  "export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64",
9
  shell=True,
 
4
  "export CPATH=$CPATH:/usr/local/cuda/include",
5
  shell=True,
6
  )
7
+ subprocess.run(
8
+ "export CUDA_HOME=/usr/local/cuda",
9
+ shell=True,
10
+ )
11
+ subprocess.run(
12
+ "export PATH=$PATH:$CUDA_HOME/bin",
13
+ shell=True,
14
+ )
15
  subprocess.run(
16
  "export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64",
17
  shell=True,