Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add CUDA 11.3 + PyTorch 1.11.0 and 1.12.1 #4205

Conversation

thiagocrepaldi
Copy link
Contributor

@thiagocrepaldi thiagocrepaldi commented May 2, 2022

Currently CI supports only CUDA 11.1, which is sufficient for PyTorch versions up to 1.10.
For PyTorch 1.11+ and nightly builds, the minimum CUDA version is 11.3.

This PR adds 3 new pipelines:

  • Linux CUDA 11.3 which installs PyTorch, TorchVision, Detectron2 nightly builds.
  • Linux CUDA 11.3 which installs PyTorch 1.11 and 1.12 stable release with matching TorchVision and Detectron2 builds.
  • Linux CPU which installs PyTorch, TorchVision, Detectron2 nightly builds.
  • Windows CPU which install PyTorch, TorchVision, Detectron2 nightly builds.

The new pipelines are important to test end-to-end ONNX export tests, in which PyTorch inference results are numerically compared with the ONNX Runtime's . A second benefit is help identifying ONNX export issues earlier in the development cycle.

ps: Nightly builds are only installed in the new pipelines (aka windows_cpu_build_pytorch_master, linux_cuda113_tests_pytorch_master_python39, linux_cpu_tests_pytorch_master. All existing pipelines are intact.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label May 2, 2022
@thiagocrepaldi thiagocrepaldi force-pushed the thiagofc/update-build-cuda branch 29 times, most recently from 9a76428 to 538226c Compare May 3, 2022 22:08
@thiagocrepaldi thiagocrepaldi requested review from wat3rBro and removed request for ppwwyyxx March 16, 2023 14:34
@thiagocrepaldi thiagocrepaldi force-pushed the thiagofc/update-build-cuda branch from 556bd03 to 360b367 Compare March 20, 2023 18:24
Thiago Crepaldi added 20 commits April 5, 2023 15:08
All non pytorch master pipelines are unchanged
3 new pileines are introduced.
* 1) Windows CPU and 2) Linux CPU with pytorch/torchvision nightly builds
* 2) Linux with CUDA 11.3, python 3.9 and nightly pytorch/torchvision

CUDA 11.3 is needed because PyTorch does not distribute wheels for
nightly 11.1

Python 3.9 was needed for the same reason; no torch wheels for 3.6
@thiagocrepaldi thiagocrepaldi force-pushed the thiagofc/update-build-cuda branch from 360b367 to c408649 Compare April 5, 2023 19:09
@thiagocrepaldi
Copy link
Contributor Author

@wat3rBro @ppwwyyxx gentle ping

@thiagocrepaldi
Copy link
Contributor Author

Is this still relevant or should we close it?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants