nasa-nccs-hpda/pytorch-caney

New 24.01 container version testing and including additional packages

Opened this issue · 0 comments

Generate and test in dev mode version 24.01 from the NVIDIA NGC pytorch container. Add the following packages:

  • pip install lightning-bolts
  • pip install deepspeed