pinned llvm-openmp version to avoid cpu affinity bug in pytorch

real-stanford · Sep 13, 2023 · 5aa9996 · 5aa9996
1 parent 5c3d54f
commit 5aa9996
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/conda_environment.yaml b/conda_environment.yaml
@@ -46,6 +46,8 @@ dependencies:
   - diffusers=0.11.1
   - av=10.0.0
   - cmake=3.24.3
+  # trick to avoid cpu affinity issue described in https://github.com/pytorch/pytorch/issues/99625
+  - llvm-openmp=14
   # trick to force reinstall imagecodecs via pip
   - imagecodecs==2022.8.8
   - pip:

diff --git a/conda_environment_real.yaml b/conda_environment_real.yaml
@@ -46,6 +46,8 @@ dependencies:
   - diffusers=0.11.1
   - av=10.0.0
   - cmake=3.24.3
+  # trick to avoid cpu affinity issue described in https://github.com/pytorch/pytorch/issues/99625
+  - llvm-openmp=14
   # trick to force reinstall imagecodecs via pip
   - imagecodecs==2022.8.8
   - pip: