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
|