build
build copied to clipboard
Update NVIDIA libs used for OSS builds
Updating to
- CUDA 11.7
- CUDNN 8.4.1
- TensorRT 8.4.1
I pushed these containers:
-
gcr.io/tensorflow-sigs/build:124-python3.10
-
gcr.io/tensorflow-sigs/build:124-python3.9
-
gcr.io/tensorflow-sigs/build:124-python3.8
-
gcr.io/tensorflow-sigs/build:124-python3.7
Re-apply the build and push to gcr.io for staging
label to rebuild and push again. This comment will only be posted once.
Thanks. Have you done any local testing with this container yet?
I have run the pip and nonpip tests locally as described in tf_sig_build_dockerfiles/README.md on V100 and A100.
The only test failures were addressed in TF PRs 56821 and 56823. All should be passing now.
These dockerfiles have just been integrated directly into the main TF repository: https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/tf_sig_build_dockerfiles
Can you recreate this PR over in the main TF repo? You can still use this PR if you want to, but we won't be able to merge it.