-
Notifications
You must be signed in to change notification settings - Fork 5.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Done] Upgrade CUDA 8.0 in Docker #1699
Conversation
@gangliao 驱动的问题可以先试试nvidia-docker,如果nvidia-docker也不行的话,应该是我们编译/镜像的问题。如果可以,估计是运行方法的问题。 |
我做了如下命令: sudo rm -rf build third_party
docker build --no-cache -t paddle-cuda-8-new .
docker run -v $PWD:/paddle -e "WITH_GPU=ON" -e "WITH_AVX=ON" paddle-cuda-8-new
docker build -t paddle-cuda-8 -f build/Dockerfile .
nvidia-docker run -it --entrypoint bash paddle-cuda-8 仍然有
git HEAD是最新的,在
git diff:
@gangliao 能否再看一下/按我贴上来的的命令也试一下? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM.应该是没问题了,暂时没有安装了cuda8的机器,在7.5的机器运行会输出:nvidia-docker | 2017/03/27 13:47:47 Error: unsupported CUDA version: driver 7.5 < image 8.0.61
说明已经正确识别了驱动了。
@helinwang @typhoonzero 可以访问试试 nvidia-docker run -it --entrypoint bash paddle-cuda-8-gang |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
我这里试了一下,还需要改一下cmake/cudnn.cmake:
diff --git a/cmake/cudnn.cmake b/cmake/cudnn.cmake
index e5b59be..c31f6c5 100644
--- a/cmake/cudnn.cmake
+++ b/cmake/cudnn.cmake
@@ -11,6 +11,7 @@ list(APPEND CUDNN_CHECK_LIBRARY_DIRS
${CUDNN_ROOT}
${CUDNN_ROOT}/lib64
${CUDNN_ROOT}/lib
+ ${CUDNN_ROOT}/lib/x86_64-linux-gnu
$ENV{CUDNN_ROOT}
$ENV{CUDNN_ROOT}/lib64
$ENV{CUDNN_ROOT}/lib
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM++
Fix #1695