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

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

Browse Source 
pinned llvm-openmp version to avoid cpu affinity bug in pytorch
This commit is contained in:
Cheng Chi
2023-09-12 23:36:52 -07:00
committed by GitHub
parent 5c3d54fca3 5aa9996fdc
commit 7dd9dc417a
2 changed files with 4 additions and 0 deletions
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: