Closed talmo closed 1 year ago
This PR adds working environment.yml files which define fully self-contained conda environments for Windows and Linux, both on CPU and GPU.
environment.yml
This is nice because it handles CUDA dependencies and their interactions with Jax so things "just work".
This was tested on Windows 10 and Ubuntu 22.04, both with GPU.
This PR adds working
environment.ymlfiles which define fully self-contained conda environments for Windows and Linux, both on CPU and GPU.This is nice because it handles CUDA dependencies and their interactions with Jax so things "just work".
This was tested on Windows 10 and Ubuntu 22.04, both with GPU.