zhiweili
commited on
Commit
·
b24e26c
1
Parent(s):
9c5b0f3
install conda
Browse files- app_onediff.py +1 -0
app_onediff.py
CHANGED
|
@@ -43,6 +43,7 @@ def runcmd(cmd, verbose = False, *args, **kwargs):
|
|
| 43 |
# options = json.loads('{"mode": "max-optimize:max-autotune:low-precision", "memory_format": "channels_last", "dynamic": true}')
|
| 44 |
# return compile_pipe(torch_module, backend="nexfort", options=options, fuse_qkv_projections=True)
|
| 45 |
|
|
|
|
| 46 |
os.system("conda install nvidia/label/cuda-12.1.0::cuda-toolkit")
|
| 47 |
print(os.environ.get('CUDA_PATH'))
|
| 48 |
print(os.environ.get('CUDA_HOME'))
|
|
|
|
| 43 |
# options = json.loads('{"mode": "max-optimize:max-autotune:low-precision", "memory_format": "channels_last", "dynamic": true}')
|
| 44 |
# return compile_pipe(torch_module, backend="nexfort", options=options, fuse_qkv_projections=True)
|
| 45 |
|
| 46 |
+
os.system("python3 -m pip install conda")
|
| 47 |
os.system("conda install nvidia/label/cuda-12.1.0::cuda-toolkit")
|
| 48 |
print(os.environ.get('CUDA_PATH'))
|
| 49 |
print(os.environ.get('CUDA_HOME'))
|