Add patch for apex to ignore minor-version mismatch of CUDA

model.py

@@ -21,6 +21,8 @@ if os.getenv('SYSTEM') == 'spaces':
         .split())
     subprocess.run('git clone https://github.com/NVIDIA/apex'.split())
     subprocess.run('git checkout 1403c21'.split(), cwd='apex')
+    with open('patch.apex') as f:
+        subprocess.run('patch -p1'.split(), cwd='apex', stdin=f)
     subprocess.run(
         'pip install -v --disable-pip-version-check --no-cache-dir --global-option --cpp_ext --global-option --cuda_ext ./'
         .split(),

patch.apex

@@ -0,0 +1,29 @@
+diff --git a/setup.py b/setup.py
+index 5f68ecf..b4d44a8 100644
+--- a/setup.py
++++ b/setup.py
+@@ -30,15 +30,15 @@ def check_cuda_torch_binary_vs_bare_metal(cuda_dir):
+     print("\nCompiling cuda extensions with")
+     print(raw_output + "from " + cuda_dir + "/bin\n")
+ 
+-    if (bare_metal_major != torch_binary_major) or (bare_metal_minor != torch_binary_minor):
+-        raise RuntimeError(
+-            "Cuda extensions are being compiled with a version of Cuda that does "
+-            "not match the version used to compile Pytorch binaries.  "
+-            "Pytorch binaries were compiled with Cuda {}.\n".format(torch.version.cuda)
+-            + "In some cases, a minor-version mismatch will not cause later errors:  "
+-            "https://github.com/NVIDIA/apex/pull/323#discussion_r287021798.  "
+-            "You can try commenting out this check (at your own risk)."
+-        )
++#    if (bare_metal_major != torch_binary_major) or (bare_metal_minor != torch_binary_minor):
++#        raise RuntimeError(
++#            "Cuda extensions are being compiled with a version of Cuda that does "
++#            "not match the version used to compile Pytorch binaries.  "
++#            "Pytorch binaries were compiled with Cuda {}.\n".format(torch.version.cuda)
++#            + "In some cases, a minor-version mismatch will not cause later errors:  "
++#            "https://github.com/NVIDIA/apex/pull/323#discussion_r287021798.  "
++#            "You can try commenting out this check (at your own risk)."
++#        )
+ 
+ 
+ def raise_if_cuda_home_none(global_option: str) -> None: