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-23 19:26:12 +00:00
|
|
|
FROM nvidia/cuda:10.1-base-ubuntu18.04
|
2020-02-26 06:50:27 +00:00
|
|
|
LABEL maintainer="Christoph Schranz <christoph.schranz@salzburgresearch.at>"
|
|
|
|
|
# The maintainers of subsequent sections may vary
|
