Skip to content

Commit 4745af9

Browse files
Merge pull request #22 from renan-siqueira/develop
Merge develop into main
2 parents 46cd35b + 3c788e3 commit 4745af9

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

Dockerfile

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
# Use a base image. In this case, we will use Python 3.11
2+
FROM python:3.11-slim-buster
3+
4+
# Define the working directory inside the container
5+
WORKDIR /app
6+
7+
# Copy the project files to the container
8+
COPY . .
9+
10+
# Install project dependencies
11+
RUN pip install --no-cache-dir -r requirements.txt
12+
13+
# Install PyTorch GPU-specific dependencies
14+
RUN pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu121/torch_nightly.html

0 commit comments

Comments
 (0)