DeepSpeed
521d329b - Fix CI issues related to cupy install (#2483)

Comment changes are shownComment changes are hidden
Commit
2 years ago
Fix CI issues related to cupy install (#2483) * remove any cupy install when setting up environments * revert previous changes to run on cu111 runners * fix for when no cupy is installed * remove cupy uninstall for workflows not using latest torch version * update to cu116 for inference tests * fix pip uninstall line * move python environment list to after DS install * remove cupy uninstall * re-add --forked * fix how we get cupy version (should be based on nvcc version)
Author
Parents
  • .github/workflows
    • File
      amd.yml
    • File
      nv-accelerate-v100.yml
    • File
      nv-inference.yml
    • File
      nv-lightning-v100.yml
    • File
      nv-nightly.yml
    • File
      nv-torch-latest-v100.yml
    • File
      nv-torch-nightly-v100.yml
    • File
      nv-torch18-p40.yml
    • File
      nv-torch18-v100.yml
    • File
      nv-transformers-v100.yml
  • File
    setup.py