Yes, I think it is a good idea.
If the contributor doesn’t make any GPU related change, they can always uninstall tf-nightly and install tf-nightly-cpu for future updates.
I don’t think we can change anything in requirements.txt. I believe it has to be the GPU version of TF to run some of the tests.
So is the large tf-nightly image itself causing any issue?
It is like either use a large image, or install a large package on startup.