Skip to content

Commit c4bec03

Browse files
committed
add 10.1 and cpu
1 parent f7296ea commit c4bec03

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

.github/workflows/build.yml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,14 @@ jobs:
4242
policy: "manylinux2014"
4343
platform: "x86_64"
4444
CUDA_BASE_IMAGE: "nvidia/cuda:10.2-cudnn7-devel-centos7"
45+
- tag-suffix: "cuda10.1"
46+
policy: "manylinux2014"
47+
platform: "x86_64"
48+
CUDA_BASE_IMAGE: "nvidia/cuda:10.1-cudnn7-devel-centos7"
49+
- tag-suffix: "cpu"
50+
policy: "manylinux2014"
51+
platform: "x86_64"
52+
CUDA_BASE_IMAGE: ""
4553

4654
env:
4755
POLICY: ${{ matrix.policy }}

build.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,8 @@ elif [ "${POLICY}" == "manylinux2014" ]; then
4848
if [ "${PLATFORM}" == "s390x" ]; then
4949
BASEIMAGE="s390x/clefos:7"
5050
else
51-
BASEIMAGE="${CUDA_BASE_IMAGE}"
51+
DEFAULT_BASEIMAGE="${MULTIARCH_PREFIX}centos:7"
52+
BASEIMAGE="${CUDA_BASE_IMAGE:-${DEFAULT_BASEIMAGE}}"
5253
fi
5354
DEVTOOLSET_ROOTPATH="/opt/rh/devtoolset-10/root"
5455
PREPEND_PATH="${DEVTOOLSET_ROOTPATH}/usr/bin:"

0 commit comments

Comments
 (0)