博客指出，在使用tensorflow/tensorflow:2.3.0-gpu Docker镜像时，尽管requirements.txt指定了tensorflow-gpu版本，但镜像内并未包含tf-gpu包。经调查，从tf2.5.0开始，GPU版本的镜像不再区分tensorflow和tensorflow-gpu，都直接支持GPU。建议更新依赖或使用合适版本的镜像。
Dockerfile：
FROM tensorflow/tensorflow:2.3.0-gpu
requirements.txt如下：
tensorflow-gpu>=2.1
执行打包过程中，正常情况下，tf版本已经符合要求，不应该重新安装，但是却一直如下：
启动tensorflow/tensorflow:2.3.0-gpu镜像，进入容器查看pip安装包：
发现里面竟然没有tf-gpu。
继续确认是否可检测GPU：
综上所述，tf-2.6.2-gpu官方镜像内部，没有tf-gpu包，内部的tf包可直接使用GPU。
emm....
调研了一下，tf2.5.0开始的gpu镜像里就不再用tensorflow-gpu了，统一都是tensorflow，且可使用GPU；
祝君顺遂
扫一扫
5331
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
