Kangarroar commited on
Commit
438c2df
1 Parent(s): d6996ca

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -2
Dockerfile CHANGED
@@ -1,10 +1,9 @@
1
- FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
2
  WORKDIR /app
3
 
4
  COPY ./requirements.txt /app/requirements.txt
5
  COPY ./packages.txt /app/packages.txt
6
  RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
7
- RUN pip install Jinja2
8
  RUN apt-get update && xargs -r -a /app/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
9
  RUN pip3 install --no-cache-dir -r /app/requirements.txt
10
 
 
1
+ FROM anibali/pytorch:1.13.0-cuda11.8
2
  WORKDIR /app
3
 
4
  COPY ./requirements.txt /app/requirements.txt
5
  COPY ./packages.txt /app/packages.txt
6
  RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
 
7
  RUN apt-get update && xargs -r -a /app/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
8
  RUN pip3 install --no-cache-dir -r /app/requirements.txt
9