Dockerfile update to fix some issues

danijar / dreamerv3

Mastering Diverse Domains through World Models

https://danijar.com/dreamerv3

MIT License

1.28k stars 219 forks source link

Dockerfile update to fix some issues #74

Closed yusuke1526 closed 5 months ago

yusuke1526 commented 1 year ago

I encountered some difficulties while trying to execute train.py within the Docker container. To resolve the problem, I made some modifications to the Dockerfile.

Changes made in the Dockerfile:

cuda version
jax version
the path of the scripts dir

Thank you for the great implementation.

ljmanso commented 12 months ago

@yusuke1526 Works like a charm!

josephrocca commented 11 months ago

I can confirm that these fixes are required to get it working as of Oct 12 2023. Note that the current Dockerfile on danijar:main flat-out doesn't work due to nvidia/cuda:11.4.2-cudnn8-devel-ubuntu20.04 not actually existing.

For the benefit of future devs/researchers trying this out, could this be merged @danijar?

danijar commented 5 months ago

I've updated the Dockerfile to Python 3.11 and CUDA 12 and verified that it's working.