Commit
·
8e2cb6b
1
Parent(s):
b2a2d2b
updated docker
Browse files- Dockerfile +3 -1
Dockerfile
CHANGED
|
@@ -2,7 +2,7 @@ FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
|
|
| 2 |
|
| 3 |
# Install system dependencies for Python and build tools
|
| 4 |
RUN apt-get update && apt-get install -y \
|
| 5 |
-
python3 python3-pip build-essential
|
| 6 |
|
| 7 |
# Create a non-root user
|
| 8 |
RUN useradd -m -u 1000 user
|
|
@@ -12,6 +12,8 @@ ENV POETRY_CACHE_DIR=/home/user/.cache/pypoetry
|
|
| 12 |
# Set the working directory to a writable location
|
| 13 |
WORKDIR /home/user/app
|
| 14 |
|
|
|
|
|
|
|
| 15 |
# Install Poetry globally as root
|
| 16 |
RUN pip install --no-cache-dir --upgrade pip \
|
| 17 |
&& pip install --no-cache-dir poetry
|
|
|
|
| 2 |
|
| 3 |
# Install system dependencies for Python and build tools
|
| 4 |
RUN apt-get update && apt-get install -y \
|
| 5 |
+
python3 python3-pip build-essential git cuda
|
| 6 |
|
| 7 |
# Create a non-root user
|
| 8 |
RUN useradd -m -u 1000 user
|
|
|
|
| 12 |
# Set the working directory to a writable location
|
| 13 |
WORKDIR /home/user/app
|
| 14 |
|
| 15 |
+
RUN nvcc --version && ls /usr/local/cuda
|
| 16 |
+
|
| 17 |
# Install Poetry globally as root
|
| 18 |
RUN pip install --no-cache-dir --upgrade pip \
|
| 19 |
&& pip install --no-cache-dir poetry
|