未验证 提交 842f56e4 编写于 作者: T tianshuo78520a 提交者: GitHub

Update cinn docker (#55888)

更新CINN docker,CI使用Ubuntu20.04,CUDA18,Python3.10环境
上级 418cc35d
......@@ -18,7 +18,7 @@ set -ex
workspace=$(cd $(dirname ${BASH_SOURCE[0]})/../..; pwd)
build_dir_name=${cinn_build:-build_cinn}
build_dir=$workspace/${build_dir_name}
py_version=${py_version:-3.8}
py_version=${py_version:-3.10}
cinn_whl_path=python/dist/cinn-0.0.0-py3-none-any.whl
......@@ -92,7 +92,7 @@ function prepare_ci {
python${py_version} -m pip install -U --no-cache-dir pip
python${py_version} -m pip install wheel
python${py_version} -m pip install sphinx==3.3.1 sphinx_gallery==0.8.1 recommonmark==0.6.0 exhale scipy breathe==4.24.0 matplotlib sphinx_rtd_theme
python${py_version} -m pip install paddlepaddle-gpu==0.0.0.post112 -f https://www.paddlepaddle.org.cn/whl/linux/gpu/develop.html
python${py_version} -m pip install paddlepaddle-gpu==0.0.0.post118 -f https://www.paddlepaddle.org.cn/whl/linux/gpu/develop.html
}
......
# Use SHA to specify the docker image to prevent the use of old cache images
# TAG: latest-dev-cuda11.2-cudnn8.2-trt8.0-gcc82
FROM registry.baidubce.com/paddlepaddle/paddle@sha256:ac757bc25c341814284ceafb274c55e36ea7dcf026a265d14f885a0fa60368f8
FROM registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.8-cudnn8.6-trt8.5-gcc82
# NVIDIA update GPG key on 04/29/2022. Fetch the public key for CI machine
# Reference: https://developer.nvidia.com/blog/updating-the-cuda-linux-gpg-repository-key/
......
# Use SHA to specify the docker image to prevent the use of old cache images
# TAG: latest-dev-cuda11.2-cudnn8.2-trt8.0-gcc82
FROM registry.baidubce.com/paddlepaddle/paddle@sha256:ac757bc25c341814284ceafb274c55e36ea7dcf026a265d14f885a0fa60368f8
FROM registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.8-cudnn8.6-trt8.5-gcc82
# NVIDIA update GPG key on 04/29/2022. Fetch the public key for CI machine
# Reference: https://developer.nvidia.com/blog/updating-the-cuda-linux-gpg-repository-key/
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册