Spaces:

yslan
/

LN3Diff_I23D

Running on Zero

NIRVANALAN commited on Aug 18

Commit

97996d9

•

1 Parent(s): a032561

update dep

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,7 +41,7 @@ def install_dependency():
     # install apex
     subprocess.run(
-        f'FORCE_CUDA=1 {sys.executable} -m pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" git+https://github.com/NVIDIA/apex.git@master',
         shell=True,
     )

     # install apex
     subprocess.run(
+        f'TORCH_CUDA_ARCH_LIST="compute capability" FORCE_CUDA=1 {sys.executable} -m pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" git+https://github.com/NVIDIA/apex.git@master',
         shell=True,
     )