Skip to content

Commit 75466f5

Browse files
feat: create Dockerfile
1 parent 13eb9e0 commit 75466f5

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)