nroggendorff commited on
Commit
4b456b9
·
verified ·
1 Parent(s): a06c1ad

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -1
Dockerfile CHANGED
@@ -1,10 +1,13 @@
1
- FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-devel
2
 
3
  WORKDIR /app
4
 
5
  COPY . .
6
 
7
  RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
 
 
 
8
 
9
  RUN pip install --no-cache-dir --upgrade pip && \
10
  pip install --no-cache-dir \
 
1
+ FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
2
 
3
  WORKDIR /app
4
 
5
  COPY . .
6
 
7
  RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
8
+ RUN apt-get install -y python3.10 python3.10-dev python3.10-distutils && \
9
+ ln -sf /usr/bin/python3.10 /usr/bin/python3 && \
10
+ curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10
11
 
12
  RUN pip install --no-cache-dir --upgrade pip && \
13
  pip install --no-cache-dir \