Spaces:
Paused
Paused
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- Dockerfile +12 -0
- 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
|