search

floydhub / dockerfiles

Deep Learning Dockerfiles
https://docs.floydhub.com/guides/environments/
Apache License 2.0
156 stars 57 forks source link

base images on cudagl (for openGL support) #43

Open AwokeKnowing opened 6 years ago

AwokeKnowing commented 6 years ago

nvidia docker version 2 eg (docker run --runtime=nvidia ...) had not worked with opengl until a few days ago. So for example, we cannot run the open ai gym environments which rely on opengl. (specifically we cannot actually render/x-forward them)

nvidia just released a new cuda 9 image with open gl. this is great news especially for people using gym environment and ROS environments.

Here you can read a bit about the issue progress https://github.com/NVIDIA/nvidia-docker/issues/534

Here's the nvidia link https://hub.docker.com/r/nvidia/cudagl/

I'm so excited about this! I'm resisting the urge to clone and rebuild all of your images just for the pytorch image! And by the way, I see you've been updating gym recently so this should go right along with it.

houqp commented 6 years ago

Yes, we are about to rebuild everything with the new cuda 9 image :)

houqp commented 6 years ago

So I looked into cudagl. Turns out it does not come with cudnn, so we can't use it for our base right now. We need to manually install cudnn into cudagl or install opengl into cuda base. I am trying to figure out if there is an easy way to do this without copy pasting.

karray commented 5 years ago

@houqp any news about cuda + cudnn + opengl?