search

rocker-org / ml

experimental machine learning container
GNU General Public License v2.0
50 stars 13 forks source link

Opencl requirements #12

Closed noamross closed 5 years ago

noamross commented 5 years ago

CUDA environment path variables are used by several packages, so it makes sense to install them in the base cuda image rather than the Tensorflow one.

I found I could install gpuR on top of cuda-dev, but not cuda. OpenCl is part of the NVIDIA cuda developer libraries so it didn't require an additional install. It did require the tweak at the bottom of the cuda-dev Dockerfile, which I found here: https://github.com/cdeterman/gpuR/issues/52.

This includes my other PR.

cboettig commented 5 years ago

Nice, thanks. and yeah, I was also thinking those R env vars are better in the cuda image than the tensorflow one. this all looks good.