OpenVINO image can't detect GPUs

[rtitle]

Hi @ravi9 @cavusmustafa,

A user brought this to my attention, thought I would run it by you. The current terra-jupyter-gatk-ovtf image downgrades tensorflow to 2.5.0 (the base image has 2.6.0 installed). It seems like tf 2.5.0 is not able to detect GPUs. Below is output from a runtime with 1 nvidia-tesla-t4 running the terra-jupyter-gatk-ovtf image:

For comparison, here is output from a runtime running the terra-jupyter-gatk image:

Question: can we safely use tensorflow 2.6.0 in terra-jupyter-gatk-ovtf? If not, we might need to use a different base image.

Let me know what you think. Thanks!

[cavusmustafa]

Hi @rtitle,
Unfortunately, current OVTF release will not work with TF 2.6.0, so terra-jupyter-gatk-ovtf might be broken if we upgrade the TF version. With the next OVTF release, we can use TF 2.6.0 safely (expected around mid November).

[rtitle]

Thanks @cavusmustafa !

Makes sense. I just tested the us.gcr.io/broad-dsp-gcr-public/terra-jupyter-gatk-ovtf:0.1.0 version and it seems to still work. So I'm proposing to peg Terra UI at 0.1.0 for now until the next OpenVINO release. See: #265

[ravi9]

Sounds good. Thanks @rtitle .