Merge pull request #21 from columbia-ai-robotics/cchi/fix_cpu_affinity

pinned llvm-openmp version to avoid cpu affinity bug in pytorch
real-stanford · Sep 13, 2023 · 7dd9dc4 · 7dd9dc4
2 parents 5c3d54f + 5aa9996
commit 7dd9dc4
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: