hongshaorou/diffusion_policy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Cheng Chi 2023-09-13 02:36:26 -04:00
parent 5c3d54fca3
commit 5aa9996fdc
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
conda_environment.yaml
View File

@ -46,6 +46,8 @@ dependencies:
- diffusers=0.11.1 - diffusers=0.11.1
- av=10.0.0 - av=10.0.0
- cmake=3.24.3 - 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 # trick to force reinstall imagecodecs via pip
- imagecodecs==2022.8.8 - imagecodecs==2022.8.8
- pip: - pip:

2
conda_environment_real.yaml
View File

@ -46,6 +46,8 @@ dependencies:
- diffusers=0.11.1 - diffusers=0.11.1
- av=10.0.0 - av=10.0.0
- cmake=3.24.3 - 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 # trick to force reinstall imagecodecs via pip
- imagecodecs==2022.8.8 - imagecodecs==2022.8.8
- pip: - pip: