cog
cog copied to clipboard
Support Tensorflow <= 1.11.0
Tensorflow <= 1.11.0 specifies CUDA as "9" but this doesn't match any image tags, because they have minor versions.
The specific problem is our version comparison doesn't match up CUDA "9" with "9.1" in CUDABaseImageFor().
Perhaps the version checking could be aware of nil minor and patch versions, so that if the minor version isn't set that "9" == "9.1".