Spaces:
Paused
Paused
Commit
·
f10b3f0
1
Parent(s):
cc56b03
Update Dockerfile
Browse files- Dockerfile +4 -4
Dockerfile
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
-
FROM
|
| 2 |
-
|
| 3 |
-
ENV PYTHONUNBUFFERED 1
|
| 4 |
-
|
| 5 |
WORKDIR /app
|
|
|
|
|
|
|
|
|
|
| 6 |
|
| 7 |
RUN export PATH="/usr/local/cuda/bin:$PATH"
|
| 8 |
RUN wget -qO- "https://cmake.org/files/v3.17/cmake-3.17.0-Linux-x86_64.tar.gz" | tar --strip-components=1 -xz -C /usr/local
|
|
|
|
| 1 |
+
FROM nvidia/cuda:11.6.0-base-ubuntu20.04
|
|
|
|
|
|
|
|
|
|
| 2 |
WORKDIR /app
|
| 3 |
+
RUN apt update && \
|
| 4 |
+
apt install --no-install-recommends -y build-essential python3 python3-pip wget && \
|
| 5 |
+
apt clean && rm -rf /var/lib/apt/lists/*
|
| 6 |
|
| 7 |
RUN export PATH="/usr/local/cuda/bin:$PATH"
|
| 8 |
RUN wget -qO- "https://cmake.org/files/v3.17/cmake-3.17.0-Linux-x86_64.tar.gz" | tar --strip-components=1 -xz -C /usr/local
|