Skip to content

Commit 83e6b39

Browse files
committed
Remove cuda 11.2 image
1 parent 8b47144 commit 83e6b39

File tree

3 files changed

+4
-5
lines changed

3 files changed

+4
-5
lines changed

build/build-image.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@ fi
3030
build_args=""
3131

3232
if [ "${image}" == "python-predictor-gpu" ]; then
33-
cuda=("10.0" "10.1" "10.1" "10.2" "10.2" "11.0" "11.1" "11.2")
34-
cudnn=("7" "7" "8" "7" "8" "8" "8" "8")
33+
cuda=("10.0" "10.1" "10.1" "10.2" "10.2" "11.0" "11.1")
34+
cudnn=("7" "7" "8" "7" "8" "8" "8")
3535
for i in ${!cudnn[@]}; do
3636
build_args="${build_args} --build-arg CUDA_VERSION=${cuda[$i]} --build-arg CUDNN=${cudnn[$i]}"
3737
docker build "$ROOT" -f $ROOT/images/$image/Dockerfile $build_args -t quay.io/cortexlabs/${image}:${CORTEX_VERSION}-cuda${cuda[$i]}-cudnn${cudnn[$i]}

build/push-image.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,8 @@ image=$1
2424
echo "$DOCKER_PASSWORD" | docker login -u "$DOCKER_USERNAME" --password-stdin
2525

2626
if [ "$image" == "python-predictor-gpu" ]; then
27-
cuda=("10.0" "10.1" "10.1" "10.2" "10.2" "11.0" "11.1" "11.2")
28-
cudnn=("7" "7" "8" "7" "8" "8" "8" "8")
27+
cuda=("10.0" "10.1" "10.1" "10.2" "10.2" "11.0" "11.1")
28+
cudnn=("7" "7" "8" "7" "8" "8" "8")
2929
for i in ${!cudnn[@]}; do
3030
docker push quay.io/cortexlabs/${image}:${CORTEX_VERSION}-cuda${cuda[$i]}-cudnn${cudnn[$i]}
3131
done

docs/workloads/dependencies/images.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ Cortex's base Docker images are listed below. Depending on the Cortex Predictor
2020
* `quay.io/cortexlabs/python-predictor-gpu:master-cuda10.2-cudnn8`
2121
* `quay.io/cortexlabs/python-predictor-gpu:master-cuda11.0-cudnn8`
2222
* `quay.io/cortexlabs/python-predictor-gpu:master-cuda11.1-cudnn8`
23-
* `quay.io/cortexlabs/python-predictor-gpu:master-cuda11.2-cudnn8`
2423
* Python Predictor (Inferentia): `quay.io/cortexlabs/python-predictor-inf:master`
2524
* TensorFlow Predictor (CPU, GPU, Inferentia): `quay.io/cortexlabs/tensorflow-predictor:master`
2625
* ONNX Predictor (CPU): `quay.io/cortexlabs/onnx-predictor-cpu:master`

0 commit comments

Comments
 (0)