目录下载tf-gpu基于拉的tf-gpu镜像构建自己的镜像启动镜像检查GPU是否可用以TensorFlow2.0为例
下载tf-gpu
在docker hub里选择要下载的tf版本(注意选带GPU和py3的)
https://hub.docker.com/r/tensorflow/tensorflow/
如:
docker pull tensorflow/tensorflow:2.0.3-gpu-py3
如果上述下载超时,可以配置清华源。
或者通过如下命令下载:
docker pull docker.m...