Skip to content

(P)ython, (S)wift,(K)eras and (J)upiter Docker container for machine and deep learning with CPU and GPU support.

License

Notifications You must be signed in to change notification settings

ChrisEngelhardt/PSKJ-Docker

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Notes

This container needs privileged permissions to run swift code.

Importent Links

Tags

Tag Info
pskj-docker:1804-cpu (default) For cpu only use. Notes: Ubuntu 18-04
pskj-docker:1804-gpu For gpu nvidia gpu support (requires nvidia-docker v2). Notes: Ubuntu 18-04,Cuda 10.0, Cudnn7

How to build yourself

cd ubunut1804-(cpu/gpu)
docker build -t pskj-docker .

Start the container (GPU)

nvidia-docker run --privileged  -p 8888:8888 chrisengelhardt/pskj-docker:1804-gpu

Start the container (CPU)

docker run --privileged  -p 8888:8888 chrisengelhardt/pskj-docker:1804-cpu

Permanently save data

To permanently save data, you need to mount a volume into the /tmp/ folder (-v)

docker run -v `pwd`/workdir:/tmp/  --privileged  -p 8888:8888 chrisengelhardt/pskj-docker:1804-cpu

About

(P)ython, (S)wift,(K)eras and (J)upiter Docker container for machine and deep learning with CPU and GPU support.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published