Spaces:
Runtime error
Runtime error
Jitesh Jain
commited on
Commit
·
91f0bb6
1
Parent(s):
7903e83
Add nvidia-smi
Browse files- Dockerfile +2 -0
Dockerfile
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
FROM python:3.8.15
|
|
|
|
|
|
|
| 2 |
|
| 3 |
RUN useradd -ms /bin/bash admin
|
| 4 |
|
|
|
|
| 1 |
FROM python:3.8.15
|
| 2 |
+
FROM nvidia/cuda:11.4-cudnn8-runtime-ubuntu18.04
|
| 3 |
+
CMD nvidia-smi
|
| 4 |
|
| 5 |
RUN useradd -ms /bin/bash admin
|
| 6 |
|