Skip to content

Commit 2c21b6d

Browse files
committed
Upgrade cuda to 8.0
1 parent e9d839d commit 2c21b6d

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# A image for building paddle binaries
22
# Use cuda devel base image for both cpu and gpu environment
3-
FROM nvidia/cuda:7.5-cudnn5-devel-ubuntu14.04
3+
FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu14.04
44
MAINTAINER PaddlePaddle Authors <[email protected]>
55

66
ARG UBUNTU_MIRROR

paddle/scripts/docker/build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ set -e
44

55
# Set BASE_IMAGE according to env variables
66
if [ ${WITH_GPU} == "ON" ]; then
7-
BASE_IMAGE="nvidia/cuda:7.5-cudnn5-runtime-ubuntu14.04"
7+
BASE_IMAGE="nvidia/cuda:8.0-cudnn5-runtime-ubuntu14.04"
88
# additional packages to install when building gpu images
99
GPU_DOCKER_PKG="python-pip python-dev"
1010
else

0 commit comments

Comments
 (0)