alessandro trinca tornidor commited on
Commit
df001cb
·
1 Parent(s): 52cbeb1

[feat] add dockerfile based on nvidia/cuda:12.3.1-runtime-ubuntu22.04 with entrypoint.sh

Browse files
Files changed (2) hide show
  1. Dockerfile +12 -0
  2. scripts/entrypoint.sh +8 -0
Dockerfile ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
2
+
3
+ WORKDIR /workspace
4
+
5
+ COPY scripts/entrypoint.sh /workspace/entrypoint.sh
6
+ RUN chmod +x /workspace/entrypoint.sh
7
+ RUN which python
8
+ RUN free -m
9
+ RUN which nvidia-smi
10
+ RUN nvidia-smi
11
+
12
+ CMD ["/workspace/entrypoint.sh"]
scripts/entrypoint.sh ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ free -m
4
+ which python
5
+ which nvidia-smi
6
+ nvidia-smi
7
+
8
+ exit 0