mirror of
https://github.com/wassname/ray.git
synced 2026-08-11 11:24:51 +08:00
[docker] Use cuDNN7, not 8 (#12375)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user