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: