Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleDetection
提交
b3b6e82b
P
PaddleDetection
项目概览
PaddlePaddle
/
PaddleDetection
大约 1 年 前同步成功
通知
695
Star
11112
Fork
2696
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
184
列表
看板
标记
里程碑
合并请求
40
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleDetection
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
184
Issue
184
列表
看板
标记
里程碑
合并请求
40
合并请求
40
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
b3b6e82b
编写于
3月 29, 2017
作者:
L
Luo Tao
浏览文件
操作
浏览文件
下载
差异文件
Merge branch 'develop' into stride
上级
5262c449
06b2e4d2
变更
7
隐藏空白更改
内联
并排
Showing
7 changed file
with
28 addition
and
17 deletion
+28
-17
Dockerfile
Dockerfile
+1
-1
cmake/cudnn.cmake
cmake/cudnn.cmake
+1
-0
doc/getstarted/build_and_install/docker_install_cn.rst
doc/getstarted/build_and_install/docker_install_cn.rst
+8
-1
doc/getstarted/build_and_install/docker_install_en.rst
doc/getstarted/build_and_install/docker_install_en.rst
+9
-2
paddle/CMakeLists.txt
paddle/CMakeLists.txt
+2
-7
paddle/scripts/docker/build.sh
paddle/scripts/docker/build.sh
+7
-5
paddle/scripts/submit_local.sh.in
paddle/scripts/submit_local.sh.in
+0
-1
未找到文件。
Dockerfile
浏览文件 @
b3b6e82b
# A image for building paddle binaries
# Use cuda devel base image for both cpu and gpu environment
FROM
nvidia/cuda:
7.5
-cudnn5-devel-ubuntu14.04
FROM
nvidia/cuda:
8.0
-cudnn5-devel-ubuntu14.04
MAINTAINER
PaddlePaddle Authors <paddle-dev@baidu.com>
ARG
UBUNTU_MIRROR
...
...
cmake/cudnn.cmake
浏览文件 @
b3b6e82b
...
...
@@ -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
...
...
doc/getstarted/build_and_install/docker_install_cn.rst
浏览文件 @
b3b6e82b
...
...
@@ -57,7 +57,14 @@ PaddlePaddle发布的docker镜像使用说明
如果输出是No,就需要选择使用no-AVX的镜像
注意:在运行GPU版本的镜像时 安装CUDA驱动,以及告诉Docker:
以上方法在GPU镜像里也能用,只是请不要忘记提前在物理机上安装GPU最新驱动。
为了保证GPU驱动能够在镜像里面正常运行,我们推荐使用[nvidia-docker](https://github.com/NVIDIA/nvidia-docker)来运行镜像。
.. code-block:: bash
nvidia-docker run -it --rm paddledev/paddle:0.10.0rc1-gpu /bin/bash
注意: 如果使用nvidia-docker存在问题,你也许可以尝试更老的方法,具体如下,但是我们并不推荐这种方法。:
.. code-block:: bash
...
...
doc/getstarted/build_and_install/docker_install_en.rst
浏览文件 @
b3b6e82b
...
...
@@ -79,8 +79,15 @@ latest versions under "tags" tab at dockerhub.com.
If it doesn't, we will use the non-AVX images.
Notice please don't forget
to install CUDA driver and let Docker knows about it:
Above methods work with the GPU image too -- just please don't forget
to install GPU driver. To support GPU driver, we recommend to use
[nvidia-docker](https://github.com/NVIDIA/nvidia-docker). Run using
.. code-block:: bash
nvidia-docker run -it --rm paddledev/paddle:0.10.0rc1-gpu /bin/bash
Note: If you would have a problem running nvidia-docker, you may try the old method we have used (not recommended).
.. code-block:: bash
...
...
paddle/CMakeLists.txt
浏览文件 @
b3b6e82b
...
...
@@ -9,13 +9,8 @@ add_subdirectory(pserver)
add_subdirectory
(
trainer
)
add_subdirectory
(
scripts
)
configure_file
(
${
CMAKE_CURRENT_SOURCE_DIR
}
/setup.py.in
${
CMAKE_CURRENT_SOURCE_DIR
}
/setup.py
)
if
(
WITH_PREDICT_SDK
)
add_subdirectory
(
predict
)
endif
()
if
(
WITH_SWIG_PY
)
configure_file
(
${
CMAKE_CURRENT_SOURCE_DIR
}
/setup.py.in
${
CMAKE_CURRENT_SOURCE_DIR
}
/setup.py
)
add_subdirectory
(
api
)
endif
()
paddle/scripts/docker/build.sh
浏览文件 @
b3b6e82b
...
...
@@ -4,7 +4,7 @@ set -e
# Set BASE_IMAGE according to env variables
if
[
${
WITH_GPU
}
==
"ON"
]
;
then
BASE_IMAGE
=
"nvidia/cuda:
7.5
-cudnn5-runtime-ubuntu14.04"
BASE_IMAGE
=
"nvidia/cuda:
8.0
-cudnn5-runtime-ubuntu14.04"
# additional packages to install when building gpu images
GPU_DOCKER_PKG
=
"python-pip python-dev"
else
...
...
@@ -12,11 +12,10 @@ else
fi
DOCKERFILE_GPU_ENV
=
""
DOCKERFILE_CUDNN_DSO
=
""
if
[[
${
WITH_GPU
:-
OFF
}
==
'ON'
]]
;
then
DOCKERFILE_GPU_ENV
=
"ENV LD_LIBRARY_PATH /usr/lib/x86_64-linux-gnu:
${
LD_LIBRARY_PATH
}
"
# for cmake to find cudnn
ln
-s
/usr/lib/x86_64-linux-gnu/libcudnn.so /usr/lib/libcudnn.so
DOCKERFILE_CUDNN_DSO
=
"RUN ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.5 /usr/lib/x86_64-linux-gnu/libcudnn.so"
fi
mkdir
-p
/paddle/build
...
...
@@ -95,7 +94,10 @@ RUN ${MIRROR_UPDATE}
# Use different deb file when building different type of images
ADD build/*.deb /usr/local/opt/paddle/deb/
# run paddle version to install python packages first
RUN dpkg -i /usr/local/opt/paddle/deb/*.deb && rm -f /usr/local/opt/paddle/deb/*.deb && paddle version
RUN dpkg -i /usr/local/opt/paddle/deb/*.deb &&
\
rm -f /usr/local/opt/paddle/deb/*.deb &&
\
paddle version
${
DOCKERFILE_CUDNN_DSO
}
${
DOCKERFILE_GPU_ENV
}
# default command shows the paddle version and exit
CMD ["paddle", "version"]
...
...
paddle/scripts/submit_local.sh.in
浏览文件 @
b3b6e82b
...
...
@@ -22,7 +22,6 @@ function version(){
echo
" with_python: @WITH_PYTHON@"
echo
" with_rdma: @WITH_RDMA@"
echo
" with_timer: @WITH_TIMER@"
echo
" with_predict_sdk: @WITH_PREDICT_SDK@"
}
function
ver2num
()
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录