Spaces:

microsoft
/

MInference

Running on Zero

iofu728 commited on Jun 17, 2024

Commit

ebc86be

1 Parent(s): f0c68fa

Fix(MInference): fix the env

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,6 +20,7 @@ subprocess.run(
 )
 subprocess.run(
     "pip install pycuda==2023.1",
     shell=True,
 )

 )
 subprocess.run(
     "pip install pycuda==2023.1",
+    env={"CPATH": "$CPATH:/usr/local/cuda/include", "LIBRARY_PATH": "$LIBRARY_PATH:/usr/local/cuda/lib64"},
     shell=True,
 )