Skip to content

H
Hydra Node Http
Commit 4e38eaac authored by kurumuz's avatar kurumuz
Browse files
Options

torch 116

parent 66fa8a20
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
Dockerfile
View file @ 4e38eaac
FROM nvidia/cuda:11.3.1-base-ubuntu20.04 FROM nvidia/cuda:11.6.1-base-ubuntu20.04
ENV DEBIAN_FRONTEND=noninteractive ENV DEBIAN_FRONTEND=noninteractive
#Install core packages #Install core packages
...@@ -11,7 +11,7 @@ RUN apt-get update && apt-get install -y \ ...@@ -11,7 +11,7 @@ RUN apt-get update && apt-get install -y \
pip3 install --no-cache-dir --upgrade pip pip3 install --no-cache-dir --upgrade pip
#Install Torch(chonky!) #Install Torch(chonky!)
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
#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
......
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