mashe commited on
Commit
1b68dfa
·
verified ·
1 Parent(s): c832f76

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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
- 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
 
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