diff --git a/paddle/scripts/README.md b/paddle/scripts/README.md index 973278d9fc4a9b1995f887e7e6867f95db429280..39db5a601d3d46c106a574870f02434bd4bd5cd1 100644 --- a/paddle/scripts/README.md +++ b/paddle/scripts/README.md @@ -112,7 +112,8 @@ We also release a script and Dockerfile for building PaddlePaddle docker images across different cuda versions. To build these docker images, run: ```bash -REPO=[your_repo_name] bash ./build_docker_images.sh +bash ./build_docker_images.sh +docker build -t [REPO]/paddle:tag -f [generated_docker_file] . ``` - NOTE: note that you can choose different base images for your environment, you can find all the versions [here](https://hub.docker.com/r/nvidia/cuda/). diff --git a/paddle/scripts/build_docker_images.sh b/paddle/scripts/build_docker_images.sh index 8779b11d6781fcf1a46779949f167004927f6487..e2b17a8fd410f554de097017bf6abd60179eb0ec 100644 --- a/paddle/scripts/build_docker_images.sh +++ b/paddle/scripts/build_docker_images.sh @@ -6,9 +6,9 @@ REPO="${REPO:-paddlepaddle}" sed 's//9.0-cudnn7-devel-ubuntu16.04/g' Dockerfile.tmp | sed 's//5.1_ga_cuda9_cudnnv7.5/g' | sed 's//2.4.7-1+cuda9.0/g' > Dockerfile.cuda9.0-cudnn7 -docker build -t ${REPO}/paddle:cuda9.0-cudnn7-devel-ubuntu16.04 -f Dockerfile.cuda9.0-cudnn7 . +# docker build -t ${REPO}/paddle:cuda9.0-cudnn7-devel-ubuntu16.04 -f Dockerfile.cuda9.0-cudnn7 . sed 's//10.0-cudnn7-devel-ubuntu16.04/g' Dockerfile.tmp | sed 's//5.1_ga_cuda10_cudnnv7.5/g' | sed 's//2.4.7-1+cuda10.0/g' > Dockerfile.cuda10.0-cudnn7 -docker build -t ${REPO}/paddle:cuda10.0-cudnn7-devel-ubuntu16.04 -f Dockerfile.cuda10.0-cudnn7 . +# docker build -t ${REPO}/paddle:cuda10.0-cudnn7-devel-ubuntu16.04 -f Dockerfile.cuda10.0-cudnn7 .