# Use NVIDIA CUDA as base image and run the same installation as in the other packages. # The version of cudatoolkit must match those of the base image, see Dockerfile.pytorch FROM nvidia/cuda:10.1-base-ubuntu18.04 LABEL maintainer="Christoph Schranz " # The maintainers of subsequent sections may vary