Share your work here (Part 2)

[ste (Stefano Giomo)]

Swift fot Tensorflow using GPU on docker

The guide contains:

• Installation guide
• Custom startup script to use gpu on docker
• A script to ssh into your container
• nvidia-smi support

Swift for tensorflow using GPU with nvidia docker

[image] INSTALL Install nvidia docker from nvidia website: https://github.com/NVIDIA/nvidia-docker. NOTE: be sure to see the result of nvidia smi on test container after install. Clone https://github.com/google/swift-jupyter and build nvidia docker image git clone https://github.com/google/swift-jupyter.git cd swift-jupyter # from inside the directory of this repository docker build -f docker/Dockerfile -t swift-jupyter . Launch docker daemon: sudo systemctl start docker RUN T…