Update Dockerfile
Browse files- Dockerfile +4 -4
Dockerfile
CHANGED
@@ -3,10 +3,10 @@ FROM python:3.9
|
|
3 |
|
4 |
# Install CUDA for GPU support (optional, based on your needs)
|
5 |
# Uncomment the following lines if you need GPU support
|
6 |
-
RUN apt-get update && apt-get install -y --no-install-recommends \
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
|
11 |
# Set the working directory in the container
|
12 |
WORKDIR /app
|
|
|
3 |
|
4 |
# Install CUDA for GPU support (optional, based on your needs)
|
5 |
# Uncomment the following lines if you need GPU support
|
6 |
+
# RUN apt-get update && apt-get install -y --no-install-recommends \
|
7 |
+
# cuda-cudart-dev-11.0 \
|
8 |
+
# && \
|
9 |
+
# rm -rf /var/lib/apt/lists/*
|
10 |
|
11 |
# Set the working directory in the container
|
12 |
WORKDIR /app
|