2020-02-20 10:03:48 +00:00
|
|
|
# Use NVIDIA CUDA as base image and run the same installation as in the other packages.
|
2020-02-22 19:35:24 +00:00
|
|
|
# The version of cudatoolkit must match those of the base image, see Dockerfile.pytorch
|
2020-02-22 19:50:43 +00:00
|
|
|
FROM nvidia/cuda:10.1-base-ubuntu18.04
|
2020-02-20 10:03:48 +00:00
|
|
|
LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"
|
