Commit dc93d33c authored by Jennifer Wei's avatar Jennifer Wei
Browse files

remove mpipy from pip install requirements

An mpi lib file is missing from the currently selected docker image. More testing is needed to support mpi mode for the docker CI.
parent 1647ec90
...@@ -33,7 +33,6 @@ dependencies: ...@@ -33,7 +33,6 @@ dependencies:
- pytorch::pytorch=2.1 - pytorch::pytorch=2.1
- pytorch::pytorch-cuda=12.1 - pytorch::pytorch-cuda=12.1
- pip: - pip:
- mpi4py==3.1.5
- deepspeed==0.12.4 - deepspeed==0.12.4
- dm-tree==0.1.6 - dm-tree==0.1.6
- git+https://github.com/NVIDIA/dllogger.git - git+https://github.com/NVIDIA/dllogger.git
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment