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
- 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:

2
conda_environment_real.yaml
View File

@ -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: