- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
-
|
| 2 |
-
FROM ubuntu:latest
|
| 3 |
WORKDIR /app
|
| 4 |
RUN mkdir -p /app && chown ubuntu:ubuntu /app && chmod 777 /app
|
| 5 |
RUN apt-get update -y && apt-get upgrade
|
|
|
|
| 1 |
+
FROM nvidia/cuda:12.8.0-cudnn-devel-ubuntu22.04
|
| 2 |
+
#FROM ubuntu:latest
|
| 3 |
WORKDIR /app
|
| 4 |
RUN mkdir -p /app && chown ubuntu:ubuntu /app && chmod 777 /app
|
| 5 |
RUN apt-get update -y && apt-get upgrade
|