Stop using tensorflow-testing docker images and start using hub.docker.com/tensorflow images

The master branch is broken because the docker image used to test on gpus is the one used to test the tensorflow source code itself. This can be considered private API.

Instead, to build our manylinux2010 wheels and test our code, we should use the official way recommended by tensorflow (the same way external users do it).

Here is the official guide: https://github.com/tensorflow/custom-op The docker images we should use are from the official dockerhub repo:

tensorflow/tensorflow:2.1.0-custom-op-gpu-ubuntu16  # gpu
tensorflow/tensorflow:2.1.0-custom-op-ubuntu16          # cpu

Those images are “manylinux2010 compatible” sort of, since it’s ubuntu.

By doing that, our testing environment will become stable again, and we shouldn’t suffer sudden build breaks.

[seanpmorgan]

I’ve tested the image locally and it’s not a network error 😦

Looks like it’s because python3.8 was added to the container.

[gabrieldemarmiesse]

Nice find @seanmorgan! I still think we should switch images before they do more drastic changes, if they change the cuda version, the build will break again.