You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
11 lines
294 B
Docker
11 lines
294 B
Docker
5 years ago
|
FROM nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04
|
||
|
|
RUN apt-get update &&\
|
||
|
|
apt-get install -y python3 python3-pip &&\
|
||
|
|
apt-get clean && rm -rf /var/lib/apt/lists/*
|
||
|
|
COPY requirements.txt /tmp
|
||
|
|
RUN pip3 install -r /tmp/requirements.txt
|
||
|
|
|
||
|
|
COPY test-nvidia /usr/bin
|
||
|
|
|
||
|
|
CMD ["/usr/bin/test-nvidia"]
|