New Envs: Pytorch 0.4.1, TF 1.10

floydhub / dockerfiles

Deep Learning Dockerfiles

https://docs.floydhub.com/guides/environments/

Apache License 2.0

156 stars 57 forks source link

New Envs: Pytorch 0.4.1, TF 1.10 #55

Closed ReDeiPirati closed 6 years ago

ReDeiPirati commented 6 years ago

hey @houqp, before merging this PR we need to rebuild the base from nvidia/cuda:9.2-cudnn7-devel-ubuntu16.04

What's new on this PR:

Added Pytorch v0.4.1
Added Torchtext on the pytorch envs
Use latest h5py in dl-base to fix python 2. compatibility issue (see this thread)

houqp commented 6 years ago

@ReDeiPirati why do we need to rebuild base from nvidia/cuda:9.2-cudnn7-devel-ubuntu16.04?

ReDeiPirati commented 6 years ago

Because the Prebuilt binaries are now built against CUDA 9.2 or CUDA 9.0 (even if it's unclear the installation step for CUDA 9.0 from the pytorch page). They have also started to support python 3.7.

ReDeiPirati commented 6 years ago

Add TF v.1.10 (+ Keras 2.2.2)