hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-05 10:01:43 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[docker] Use cuDNN7, not 8 (#12375)

Browse source
This commit is contained in:
Ian Rodney 2020-11-25 12:06:53 -08:00 • committed by GitHub
parent 753cda2f28
commit ca6c2b2442
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
build-docker.sh
View file

@ -16,7 +16,7 @@ key="$1"
case $key in
--gpu)
GPU="-gpu"
BASE_IMAGE="nvidia/cuda:10.1-cudnn8-runtime-ubuntu18.04"
BASE_IMAGE="nvidia/cuda:10.1-cudnn7d-runtime-ubuntu18.04"
;;
--no-cache-build)
NO_CACHE="--no-cache"

2
ci/travis/build-docker-images.py
View file

@ -84,7 +84,7 @@ def _build_cpu_gpu_images(image_name, no_cache=True) -> List[str]:
build_args = {}
if image_name == "base-deps":
build_args["BASE_IMAGE"] = (
"nvidia/cuda:10.1-cudnn8-runtime-ubuntu18.04"
"nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04"
if gpu == "-gpu" else "ubuntu:focal")
else:
build_args["GPU"] = gpu

2
ci/travis/build-docker-images.sh
View file

@ -22,7 +22,7 @@ build_and_push_tags() {
# $2 tag for image (e.g. hash of commit)
for GPU in "" "-gpu"
do
BASE_IMAGE=$(if [ "$GPU" ]; then echo "nvidia/cuda:10.1-cudnn8-runtime-ubuntu18.04"; else echo "ubuntu:focal"; fi;)
BASE_IMAGE=$(if [ "$GPU" ]; then echo "nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04"; else echo "ubuntu:focal"; fi;)
FULL_NAME_WITH_TAG="rayproject/$1:$2$GPU"
NIGHTLY_FULL_NAME_WITH_TAG="rayproject/$1:nightly$GPU"
docker build --no-cache --build-arg GPU="$GPU" --build-arg BASE_IMAGE="$BASE_IMAGE" --build-arg WHEEL_PATH=".whl/$WHEEL" --label "SHA=$2" -t "$FULL_NAME_WITH_TAG" /"$ROOT_DIR"/docker/"$1"

2
docker/base-deps/Dockerfile
View file

@ -1,6 +1,6 @@
# The base-deps Docker image installs main libraries needed to run Ray
# The GPU option is nvidia/cuda:10.1-cudnn8-runtime-ubuntu18.04
# The GPU option is nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04
ARG BASE_IMAGE="ubuntu:focal"
FROM ${BASE_IMAGE}
# If this arg is not "autoscaler" then no autoscaler requirements will be included