Dockerfile doesn't work anymore - Githubissues

google / dopamine

Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.

https://github.com/google/dopamine

Apache License 2.0

10.42k stars 1.36k forks source link

Dockerfile doesn't work anymore #188

Open SergioArnaud opened 2 years ago

SergioArnaud commented 2 years ago

I had several issues running dopamine with the last docker image because of the last Jax and CUDA changes. Particularly with the cudnn versions.

I made the following changes to the dockerfile and managed to the image successfully in a gpu:

Changed this line with ARG cuda_docker_tag="11.4.2-cudnn8-devel-ubuntu20.04" to update the docker cuda image to 11.4.2:
Changed this line to RUN pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_releases.html since the installation command changed.