Install libstdcxx-ng=12.3.0

pytorch-labs · Nov 21, 2024 · 8a8a5aa · 8a8a5aa
1 parent a5c5020
commit 8a8a5aa
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tools/cuda_utils.py b/tools/cuda_utils.py
@@ -129,6 +129,12 @@ def install_torch_deps(cuda_version: str):
     ]
     cmd = ["conda", "install", "-y"] + torch_deps
     subprocess.check_call(cmd)
+    # conda forge deps
+    # ubuntu 22.04 comes with libstdcxx6 12.3.0
+    # we need to install the same library version in conda
+    conda_deps = ["libstdcxx-ng=12.3.0"]
+    cmd = ["conda", "install", "-y", "-c", "conda-forge"] + conda_deps
+    subprocess.check_call(cmd)
 
 
 def install_torch_build_deps(cuda_version: str):