feat: Upgrade Docker build to use custom TRT + CUDNN by gs-olive · Pull Request #1805 · pytorch/TensorRT

@gs-olive

- Upgrade Ubuntu version to 22.04
- Make custom build args for TensorRT and CUDNN versions,
user-specifiable in a.b.c version format
- Make build args required, without defaults, to improve code
cleanliness and reduce amount of changes needed when versions shift
- Provide clear error messages when the build args are not provided by
the user
- Add documentation of the change in the Docker README

@gs-olive

@gs-olive gs-olive added WIP

Work is in progress, pull request should not be merged yet

and removed WIP

Work is in progress, pull request should not be merged yet

labels

Apr 6, 2023

gs-olive

gs-olive

@gs-olive

- Add support for pyenv in Docker file, allowing specification of a
custom Python version
- Update WORKSPACE files to automatically synchronize regardless of
Python version choice, using symbolic links
- Update dependencies accordingly
- Add documentation for changing the base image of the Docker container

narendasan

@gs-olive

- Add documentation regarding Python versioning

bowang007 pushed a commit that referenced this pull request

Apr 28, 2023

@gs-olive @bowang007