NVIDIA / modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

https://developer.nvidia.com/modulus

Apache License 2.0

836 stars 190 forks source link

Update library versions (PyTorch base container and ONNX) #411

Closed ktangsali closed 3 months ago

ktangsali commented 3 months ago

Modulus Pull Request

Description

Updated the PyTorch base container in preparation for the upcoming release. ONNX needs an update too due to the following bug that was fixed in this PR: https://github.com/microsoft/onnxruntime/pull/19419/files

Checklist

[x] I am familiar with the Contributing Guidelines.
[x] New or existing tests cover these changes.
[x] The documentation is up to date with these changes.
[ ] The CHANGELOG.md is up to date with these changes.
[ ] An issue is linked to this pull request.

Dependencies

Alexey-Kamenev commented 3 months ago

/blossom-ci