We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e9d839d commit 2c21b6dCopy full SHA for 2c21b6d
Dockerfile
@@ -1,6 +1,6 @@
1
# A image for building paddle binaries
2
# Use cuda devel base image for both cpu and gpu environment
3
-FROM nvidia/cuda:7.5-cudnn5-devel-ubuntu14.04
+FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu14.04
4
MAINTAINER PaddlePaddle Authors <[email protected]>
5
6
ARG UBUNTU_MIRROR
paddle/scripts/docker/build.sh
@@ -4,7 +4,7 @@ set -e
# Set BASE_IMAGE according to env variables
if [ ${WITH_GPU} == "ON" ]; then
7
- BASE_IMAGE="nvidia/cuda:7.5-cudnn5-runtime-ubuntu14.04"
+ BASE_IMAGE="nvidia/cuda:8.0-cudnn5-runtime-ubuntu14.04"
8
# additional packages to install when building gpu images
9
GPU_DOCKER_PKG="python-pip python-dev"
10
else
0 commit comments