Where is the DockerFile for building custom Tensorflow

TensorFlow General Discussion
,
1

Hi, I wonder how can I create a docker image match with this one (TensorFlow | NVIDIA NGC). Or is there any tutorial that I can build tensorflow by myself that support newer cuda version (12.x) and newer tensorrt (8.6.x).

3

@Minh_Anh,

Welcome to the Tensorflow Forum!

You can create a Tensorflow docker container by following the instructions in the TF SIG Build docker files guide.

Thank you!