wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-15 12:45:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Docker] Use Cuda 11 (#13691)

Browse Source
This commit is contained in:
Ian Rodney
2021-01-27 13:45:30 -08:00
committed by GitHub
parent eba698d48e
commit b4bcb9b60a
6 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/base-deps/Dockerfile
+1 -1
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-cudnn7-runtime-ubuntu18.04
# The GPU option is nvidia/cuda:11.0-cudnn8-runtime-ubuntu18.04
ARG BASE_IMAGE="ubuntu:focal"
FROM ${BASE_IMAGE}
# If this arg is not "autoscaler" then no autoscaler requirements will be included