Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit b0c34b0

Browse files
author
Jonathan Hseu
committed
Workaround for cudnn file location issue
1 parent 49b8223 commit b0c34b0

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tensorflow/tools/ci_build/Dockerfile.gpu

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,11 @@ FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu14.04
22

33
MAINTAINER Jan Prach <[email protected]>
44

5+
# In the Ubuntu 14.04 images, cudnn is placed in system paths. Move them to
6+
# /usr/local/cuda
7+
RUN cp /usr/include/cudnn.h /usr/local/cuda/include
8+
RUN cp /usr/lib/x86_64-linux-gnu/libcudnn* /usr/local/cuda/lib64
9+
510
# Copy and run the install scripts.
611
COPY install/*.sh /install/
712
RUN /install/install_bootstrap_deb_packages.sh

0 commit comments

Comments
 (0)