search

rocker-org / ml

experimental machine learning container
GNU General Public License v2.0
50 stars 13 forks source link

Additional cuda libs for tensorflow, based on tensorflow/tensorflow gpu image #19

Closed cboettig closed 5 years ago

cboettig commented 5 years ago

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/dockerfiles/dockerfiles/gpu.Dockerfile