0.6.0 docker patch

[ktangsali]

Modulus Pull Request

Description

With different base containers, we have different requirements for the wheels. This PR allows a mechanism to control this using a build-arg.

Also tested the build using just the defaults (in the absence of pre-built wheels)

Closes: https://github.com/NVIDIA/modulus/issues/453

Checklist

• [x] I am familiar with the Contributing Guidelines.
• [x] New or existing tests cover these changes.
• [ ] 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

[ktangsali]

/blossom-ci

[ktangsali]

This looks good to me. Just to confirm, this PR should now allow local docker builds without having the cached wheels right?

Yes, that is correct. There is one caveat, if the base container is changed such that the fallbacks don't work (say has unsupported CUDA version, etc.), then that might require some additional handling, but other than that, it should work fine. If you run into issues, please let me know. I tested this dockerfile without any cached wheels (just defaults, i.e. make container-deploy) and that worked.