Skip to content

Commit 4a57870

Browse files
committed
Added experimental Dockerfile support
1 parent 7f8f66d commit 4a57870

1 file changed

Lines changed: 17 additions & 0 deletions

File tree

Dockerfile

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
FROM nvidia/cuda:latest
2+
3+
# packaging dependencies
4+
RUN apt update -y && apt install git-core wget -y
5+
RUN git clone https://github.com/gpue-group/gpue /gpue && \
6+
wget https://cmake.org/files/v3.13/cmake-3.13.0-rc2-Linux-x86_64.sh && \
7+
chmod +x ./cmake-3.13.0-rc2-Linux-x86_64.sh && \
8+
./cmake-3.13.0-rc2-Linux-x86_64.sh --skip-license --prefix=/usr/local
9+
10+
RUN cd /gpue && cmake . && make
11+
12+
#The Docker image that is built with the above commands can be run as follows:
13+
# (sudo) docker run --runtime=nvidia --rm <IMAGE TAG> /gpue/gpue -u
14+
# It is assumed the the Nvidia docker runtime has been installed, following
15+
# the instructins provided here: https://github.com/NVIDIA/nvidia-docker
16+
17+

0 commit comments

Comments
 (0)