File tree Expand file tree Collapse file tree 3 files changed +4
-5
lines changed
docs/workloads/dependencies Expand file tree Collapse file tree 3 files changed +4
-5
lines changed Original file line number Diff line number Diff line change 3030build_args=" "
3131
3232if [ " ${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]}
Original file line number Diff line number Diff line change @@ -24,8 +24,8 @@ image=$1
2424echo " $DOCKER_PASSWORD " | docker login -u " $DOCKER_USERNAME " --password-stdin
2525
2626if [ " $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
Original file line number Diff line number Diff 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 `
You can’t perform that action at this time.
0 commit comments