Skip to content
This repository was archived by the owner on Jan 14, 2025. It is now read-only.

Commit

Permalink
Update Dockerfile.cuda
Browse files Browse the repository at this point in the history
  • Loading branch information
jonafeucht committed Jun 13, 2024
1 parent ee59bb6 commit 4cc2226
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions Dockerfile.cuda
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,10 @@ FROM nvidia/cuda:12.5.0-runtime-ubuntu22.04
WORKDIR /app
COPY . /app
RUN apt-get update && \
apt-get install -y python3-pip && \
pip install --upgrade pip
apt-get install -y software-properties-common && \
add-apt-repository universe && \
apt-get update && \
apt-get install -y python3.12 python3-pip && \
python3 -m pip install --upgrade pip
RUN pip install --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu121
CMD ["fastapi", "run", "main.py", "--proxy-headers", "--host", "0.0.0.0", "--port", "8000"]

0 comments on commit 4cc2226

Please sign in to comment.