Skip to content

KnowledgePending/Swift-TensorFlow-with-CUDA-Docker

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Docker Pulls

Image details

  • Ubuntu 18.04 Bionic
  • CUDA 10.0
  • Swift 5.0
  • TensorFlow for Swift v0.4

Required

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

Example Programs

Further Details