A simple container for using MuJoCo-mjx, and Jax on the GPU of your PC. Installs Cuda and CudNN as well as other python dependancies.
The only requirment for using this repository is Singularity/Apptainer Version >= 3.5.
Once Singularity/Apptainer has been installed, you can simply build the container using /.build.sh (This will take a while depending on your machine and wifi connection, approximately 30 minutes) and then you can launch the container using /.run.sh.
If you need to admin priveledges inside the container to isntall something, you can use /.write.sh.
- Repository is hard coded to download Cuda 12.4 and Cudnn 9.5. I will make this flexible once I figure that out.
- Some possible issue with jax installation. It seems to be using the GPU but there is a warning about a version mismatch between cuda and XLA.
- Add paths in post script
- Make CUDA and CuDDn installations dynamic.
- Look into fixing XLA and Jax mismatch?
- Make some nice mjx-examples and have this repo clone them automatically.