forked from nv-tlabs/GSCNN
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
31 lines (25 loc) · 877 Bytes
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
FROM pytorch/pytorch:1.0-cuda10.0-cudnn7-devel
RUN apt-get -y update
RUN apt-get -y upgrade
RUN apt-get update \
&& apt-get install -y software-properties-common wget \
&& add-apt-repository -y ppa:ubuntu-toolchain-r/test \
&& apt-get update \
&& apt-get install -y make git curl vim vim-gnome
# Install apt-get
RUN apt-get install -y python3-pip python3-dev vim htop python3-tk pkg-config
RUN pip3 install --upgrade pip==9.0.1
# Install from pip
RUN pip3 install pyyaml \
scipy==1.1.0 \
numpy \
tensorflow \
scikit-learn \
scikit-image \
matplotlib \
opencv-python \
torch==1.0.0 \
torchvision==0.2.0 \
torch-encoding==1.0.1 \
tensorboardX \
tqdm