AYO/dl_bench
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add TF1 platform, docker and README

Browse source
This commit is contained in:
Corentin 2021-10-05 10:51:18 +09:00
commit dbe5490c5b
28 changed files with 655 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

5
docker/pytorch.dockerfile Normal file
View file

@ -0,0 +1,5 @@
FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime
RUN pip install matplotlib seaborn && mkdir /work
WORKDIR /work

5
docker/tensorflow.dockerfile Normal file
View file

@ -0,0 +1,5 @@
FROM tensorflow/tensorflow:latest-gpu
RUN pip install matplotlib seaborn && mkdir /work
WORKDIR /work

5
docker/tensorflow_v1.dockerfile Normal file
View file

@ -0,0 +1,5 @@
FROM tensorflow/tensorflow:1.15.5-gpu
RUN pip install matplotlib seaborn && mkdir /work
WORKDIR /work