Skip to content

H
Hydra Node Http
Commit 6f83e727 authored by novelailab's avatar novelailab
Browse files
Options

fix

parent 28195f0f
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
Dockerfile
View file @ 6f83e727
FROM nvidia/cuda:11.3.1-base-ubuntu20.04 FROM nvidia/cuda:11.3.1-base-ubuntu20.04
ENV DEBIAN_FRONTEND=noninteractive ENV DEBIAN_FRONTEND=noninteractive
#Install Torch(chonky!)
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
#Install core packages #Install core packages
RUN apt-get update && apt-get install -y \ RUN apt-get update && apt-get install -y \
libncurses5 python3 python3-pip curl git apt-utils ssh ca-certificates \ libncurses5 python3 python3-pip curl git apt-utils ssh ca-certificates \
...@@ -13,6 +10,9 @@ RUN apt-get update && apt-get install -y \ ...@@ -13,6 +10,9 @@ RUN apt-get update && apt-get install -y \
update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1 && \ update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1 && \
pip3 install --no-cache-dir --upgrade pip pip3 install --no-cache-dir --upgrade pip
#Install Torch(chonky!)
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
#Install Python deps #Install Python deps
RUN pip3 install --no-cache-dir dotmap icecream sentry-sdk numpy fastapi "uvicorn[standard]" gunicorn RUN pip3 install --no-cache-dir dotmap icecream sentry-sdk numpy fastapi "uvicorn[standard]" gunicorn
RUN pip3 install omegaconf transformers einops RUN pip3 install omegaconf transformers einops
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment