Create Tensorflow CPU base image

mlbench / mlbench-benchmarks

Distributed ML Training Benchmarks

https://mlbench.github.io

Apache License 2.0

27 stars 6 forks source link

Create Tensorflow CPU base image #13

Open Panaetius opened 5 years ago

Panaetius commented 5 years ago

Tensorflow by default depends on CUDA, even when no GPUs are utilized. We need a base image with tensorflow compiled without CUDA to run CPU experiments on nodes without CUDA installed.

martinjaggi commented 5 years ago

low prio, move to 3.0