From 5aa9996fdc9cbf1c7567057e94e33f997515bddd Mon Sep 17 00:00:00 2001 From: Cheng Chi Date: Wed, 13 Sep 2023 02:36:26 -0400 Subject: [PATCH] pinned llvm-openmp version to avoid cpu affinity bug in pytorch --- conda_environment.yaml | 2 ++ conda_environment_real.yaml | 2 ++ 2 files changed, 4 insertions(+) diff --git a/conda_environment.yaml b/conda_environment.yaml index 97cbe9e2..c540b5bd 100644 --- 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 index 1f845990..76363607 100644 --- 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: