Add GPU/ MAC acceleration to docker container

TeachMeTW / CNN

https://www.cisconeural.net

4 stars 2 forks source link

Open dleiferives opened 2 weeks ago

dleiferives commented 2 weeks ago

Current docker trains using cpu by default and installs tensorflow as default. Configure docker to run with tf for mac or nvidia or normal cpu.