Skip to content

Mohamedgalil/tf_openailab_gpu_docker

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CUDA + TensorFlow + OpenAI Gym + RoboSchool + Jupyter

This project hosts a Dockerimage that let's you run TensorFlow code to compete in the OpenAI Gym while staying true to FOSS and not relying on MujoCo. And because we are all beginners here, let's not assume our Python code is perfect from the start --> Jupyter to the rescue

Size of the image

This image is quiet large. It's because it contains

  • CUDA
  • FFMPEG + bunch of dependencies for Roboschool
  • Lots of Python packages
  • x11 xvfb etc so we can SEE the action with VNC
  • gcc and g++ compilers for C code
  • is Ubuntu based (taken from NVIDIA, I know, an alpine base would be nice)

Dependencies

You just need a machine with Docker, Nvidia-Docker and the nvidia drivers. This might also be a cloud machine, but be careful they cost quiet a bit per hour with a GPU ;-)

Ah and one more thing. Sorry, I can't pack this, as it's protected by NVIDIA lawyer blabla Download the cuDNN library files and place them in the same folder as the build.sh and Dockerfile files.

Download from here make sure you get the version cudnn-8.0-linux-x64-v7.tgz

installation

#builds and runs the image, binding your path to the container, so you can access them from jupyter
./build.sh path/to/your/notebooks 

Important links

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Shell 52.5%
  • Dockerfile 47.5%