zdou0830 commited on
Commit
ddd5fa5
1 Parent(s): ec63fb5
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -4,6 +4,9 @@ from pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel
4
  #FROM pengchuanzhang/maskrcnn:ubuntu18-py3.7-cuda10.2-pytorch1.9
5
 
6
 
 
 
 
7
  WORKDIR /code
8
 
9
  COPY ./requirements.txt /code/requirements.txt
 
4
  #FROM pengchuanzhang/maskrcnn:ubuntu18-py3.7-cuda10.2-pytorch1.9
5
 
6
 
7
+ RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
8
+
9
+
10
  WORKDIR /code
11
 
12
  COPY ./requirements.txt /code/requirements.txt