Skip to content

Commit dca5f74

Browse files
authored
Merge pull request #16 from fatihbaltaci/master
Add dockerfile
2 parents 1f6c31f + f871492 commit dca5f74

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

Dockerfile

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
FROM pytorch/pytorch:1.5-cuda10.1-cudnn7-runtime
2+
3+
RUN apt-get update && \
4+
apt-get install -y \
5+
git && \
6+
rm -rf /var/cache/apk/*
7+
8+
COPY requirements.txt /workspace
9+
10+
RUN pip --no-cache-dir install -r /workspace/requirements.txt
11+
12+
RUN pip --no-cache-dir install 'git+https://github.com/paperswithcode/torchbench.git'

0 commit comments

Comments
 (0)