File size: 212 Bytes
8d9de2d
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
FROM pytorch/pytorch:1.9.1-cuda11.1-cudnn8-runtime

WORKDIR /usr/src/app

RUN apt-get update 
RUN apt-get install libgl1 -y

COPY requirements.txt ./
RUN pip install --no-cache-dir -r requirements.txt

COPY . .