File tree Expand file tree Collapse file tree 6 files changed +16
-9
lines changed Expand file tree Collapse file tree 6 files changed +16
-9
lines changed Original file line number Diff line number Diff line change @@ -249,7 +249,7 @@ def BuildCUDA(args) {
249
249
sh """
250
250
${ dockerRun} ${ container_type} ${ docker_binary} ${ docker_args} tests/ci_build/build_via_cmake.sh -DUSE_CUDA=ON -DUSE_NCCL=ON -DOPEN_MP:BOOL=ON
251
251
${ dockerRun} ${ container_type} ${ docker_binary} ${ docker_args} bash -c "cd python-package && rm -rf dist/* && python setup.py bdist_wheel --universal"
252
- ${ dockerRun} ${ container_type} ${ docker_binary} ${ docker_args} python3 tests/ci_build/rename_whl.py python-package/dist/*.whl ${ commit_id} manylinux1_x86_64
252
+ ${ dockerRun} ${ container_type} ${ docker_binary} ${ docker_args} python3 tests/ci_build/rename_whl.py python-package/dist/*.whl ${ commit_id} manylinux2010_x86_64
253
253
"""
254
254
// Stash wheel for CUDA 9.0 target
255
255
if (args. cuda_version == ' 9.0' ) {
Original file line number Diff line number Diff line change @@ -3,7 +3,7 @@ FROM nvidia/cuda:$CUDA_VERSION-devel-centos6
3
3
4
4
# Environment
5
5
ENV DEBIAN_FRONTEND noninteractive
6
- ENV DEVTOOLSET_URL_ROOT http://mirror .centos.org/centos/6 /sclo/x86_64/rh/devtoolset-4
6
+ ENV DEVTOOLSET_URL_ROOT http://vault .centos.org/6.9 /sclo/x86_64/rh/devtoolset-4/
7
7
8
8
# Install all basic requirements
9
9
RUN \
Original file line number Diff line number Diff line change 1
1
FROM centos:6
2
2
3
+ ENV DEVTOOLSET_URL_ROOT http://vault.centos.org/6.9/sclo/x86_64/rh/devtoolset-4/
4
+
3
5
# Install all basic requirements
4
6
RUN \
5
7
yum -y update && \
6
8
yum install -y tar unzip wget xz git centos-release-scl yum-utils java-1.8.0-openjdk-devel && \
7
9
yum-config-manager --enable centos-sclo-rh-testing && \
8
10
yum -y update && \
9
- yum install -y devtoolset-6-gcc devtoolset-6-binutils devtoolset-6-gcc-c++ && \
11
+ yum install -y $DEVTOOLSET_URL_ROOT/devtoolset-4-gcc-5.3.1-6.1.el6.x86_64.rpm \
12
+ $DEVTOOLSET_URL_ROOT/devtoolset-4-gcc-c++-5.3.1-6.1.el6.x86_64.rpm \
13
+ $DEVTOOLSET_URL_ROOT/devtoolset-4-binutils-2.25.1-8.el6.x86_64.rpm \
14
+ $DEVTOOLSET_URL_ROOT/devtoolset-4-runtime-4.1-3.sc1.el6.x86_64.rpm \
15
+ $DEVTOOLSET_URL_ROOT/devtoolset-4-libstdc++-devel-5.3.1-6.1.el6.x86_64.rpm && \
10
16
# Python
11
17
wget -O Miniconda3.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
12
18
bash Miniconda3.sh -b -p /opt/python && \
19
25
ln -s /opt/apache-maven-3.6.1/ /opt/maven
20
26
21
27
ENV PATH=/opt/python/bin:/opt/maven/bin:$PATH
22
- ENV CC=/opt/rh/devtoolset-6 /root/usr/bin/gcc
23
- ENV CXX=/opt/rh/devtoolset-6 /root/usr/bin/c++
24
- ENV CPP=/opt/rh/devtoolset-6 /root/usr/bin/cpp
28
+ ENV CC=/opt/rh/devtoolset-4 /root/usr/bin/gcc
29
+ ENV CXX=/opt/rh/devtoolset-4 /root/usr/bin/c++
30
+ ENV CPP=/opt/rh/devtoolset-4 /root/usr/bin/cpp
25
31
26
32
# Install Python packages
27
33
RUN \
Original file line number Diff line number Diff line change 10
10
vcomp140_path = 'C:\\ Windows\\ System32\\ vcomp140.dll'
11
11
12
12
for wheel_path in sorted (glob .glob (sys .argv [1 ])):
13
- m = re .search (r'xgboost-(.*)-py2. py3' , wheel_path )
13
+ m = re .search (r'xgboost-(.*)-py3' , wheel_path )
14
14
assert m
15
15
version = m .group (1 )
16
16
Original file line number Diff line number Diff line change @@ -21,7 +21,7 @@ def cd(path):
21
21
commit_id = sys .argv [2 ]
22
22
platform_tag = sys .argv [3 ]
23
23
24
- assert platform_tag in ['manylinux1_x86_64' , 'win_amd64' ]
24
+ assert platform_tag in ['manylinux1_x86_64' , 'manylinux2010_x86_64' , ' win_amd64' ]
25
25
26
26
dirname , basename = os .path .dirname (whl_path ), os .path .basename (whl_path )
27
27
@@ -32,6 +32,6 @@ def cd(path):
32
32
'version' : tokens [1 ],
33
33
'commit_id' : commit_id ,
34
34
'platform_tag' : platform_tag }
35
- new_name = '{pkg_name}-{version}+{commit_id}-py2. py3-none-{platform_tag}.whl' .format (** keywords )
35
+ new_name = '{pkg_name}-{version}+{commit_id}-py3-none-{platform_tag}.whl' .format (** keywords )
36
36
print ('Renaming {} to {}...' .format (basename , new_name ))
37
37
os .rename (basename , new_name )
Original file line number Diff line number Diff line change 7
7
# Install XGBoost Python package
8
8
function install_xgboost {
9
9
wheel_found=0
10
+ pip install --upgrade pip --user
10
11
for file in python-package/dist/* .whl
11
12
do
12
13
if [ -e " ${file} " ]
You can’t perform that action at this time.
0 commit comments