- Ubuntu 18.04 Bionic
- CUDA 10.0
- Swift 5.0
- TensorFlow for Swift v0.4
Option 1. Build Docker Image
- From within the directory of the Dockerfile execute the following command to build the image
docker build -t swift_tf_cuda_docker .
- To run with bash and a shared volume
docker run --runtime=nvidia -v <host_path>:<container_path> -ti swift_tf_cuda_docker:latest bash
Option 2. Pull image from Docker Hub
docker pull bryankp/swift_tf_cuda:latest
- To run with bash and a shared volume
docker run --runtime=nvidia -v <host_path>:<container_path> -ti bryankp/swift_tf_cuda_docker:latest bash