6 lines
342 B
Docker
6 lines
342 B
Docker
# 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 <christoph.schranz@salzburgresearch.at>"
|
|
# The maintainers of subsequent sections may vary
|