未验证 提交 52edaecc 编写于 作者: P pangyoki 提交者: GitHub

modify dockerfile: support cuda11 and delete gcc8.2 in cpu version (#30746)

* support cuda11 and delete gcc8.2 in cpu version

* change method

* fix pip

* change 11 to 11.0
上级 9b3c80c8
......@@ -41,8 +41,8 @@ RUN apt-get update && \
python3.6 python3.6-dev \
python3.7 python3.7-dev \
python3.8 python3.8-dev && \
curl https://bootstrap.pypa.io/ez_setup.py -o - | python2.7 && easy_install pip && \
curl https://bootstrap.pypa.io/ez_setup.py -o - | python3.5 && easy_install pip && \
curl https://bootstrap.pypa.io/2.7/get-pip.py -o - | python2.7 && easy_install pip && \
curl https://bootstrap.pypa.io/3.5/get-pip.py -o - | python3.5 && easy_install pip && \
curl https://bootstrap.pypa.io/ez_setup.py -o - | python3.6 && easy_install pip && \
curl https://bootstrap.pypa.io/ez_setup.py -o - | python3.7 && easy_install pip && \
curl https://bootstrap.pypa.io/ez_setup.py -o - | python3.8 && easy_install pip && \
......
......@@ -42,8 +42,8 @@ RUN apt-get update && \
python3.6 python3.6-dev \
python3.7 python3.7-dev \
python3.8 python3.8-dev python3.8-distutils && \
curl https://bootstrap.pypa.io/ez_setup.py -o - | python2.7 && easy_install pip && \
curl https://bootstrap.pypa.io/ez_setup.py -o - | python3.5 && easy_install pip && \
curl https://bootstrap.pypa.io/2.7/get-pip.py -o - | python2.7 && easy_install pip && \
curl https://bootstrap.pypa.io/3.5/get-pip.py -o - | python3.5 && easy_install pip && \
curl https://bootstrap.pypa.io/ez_setup.py -o - | python3.6 && easy_install pip && \
curl https://bootstrap.pypa.io/ez_setup.py -o - | python3.7 && easy_install pip && \
curl https://bootstrap.pypa.io/ez_setup.py -o - | python3.8 && easy_install pip && \
......
......@@ -32,11 +32,15 @@ function ref_whl(){
ref_mkl=openblas
fi
if [[ ${gcc_version} == "8.2.0" ]];then
if [[ ${WITH_GPU} != "ON" ]]; then
ref_gcc = ""
elif [[ ${gcc_version} == "8.2.0" ]];then
ref_gcc=_gcc8.2
fi
if [[ ${ref_CUDA_MAJOR} == "10" ]];then
if [[ ${ref_CUDA_MAJOR} == "11.0" ]];then
ref_version=.post110
elif [[ ${ref_CUDA_MAJOR} == "10" ]];then
ref_version=.post100
elif [[ ${ref_CUDA_MAJOR} == "10.1" ]];then
ref_version=.post101
......
......@@ -32,11 +32,15 @@ function ref_whl(){
ref_mkl=openblas
fi
if [[ ${gcc_version} == "8.2.0" ]];then
if [[ ${WITH_GPU} != "ON" ]]; then
ref_gcc = ""
elif [[ ${gcc_version} == "8.2.0" ]];then
ref_gcc=_gcc8.2
fi
if [[ ${ref_CUDA_MAJOR} == "10" ]];then
if [[ ${ref_CUDA_MAJOR} == "11.0" ]];then
ref_version=.post110
elif [[ ${ref_CUDA_MAJOR} == "10" ]];then
ref_version=.post100
elif [[ ${ref_CUDA_MAJOR} == "10.1" ]];then
ref_version=.post101
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册