mirror of
https://github.com/wassname/ray.git
synced 2026-08-14 12:40:23 +08:00
fix gpu base image name in build-docker.sh script (#12642)
This commit is contained in:
+1
-1
@@ -16,7 +16,7 @@ key="$1"
|
||||
case $key in
|
||||
--gpu)
|
||||
GPU="-gpu"
|
||||
BASE_IMAGE="nvidia/cuda:10.1-cudnn7d-runtime-ubuntu18.04"
|
||||
BASE_IMAGE="nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04"
|
||||
;;
|
||||
--no-cache-build)
|
||||
NO_CACHE="--no-cache"
|
||||
|
||||
Reference in New Issue
Block a user