mirror of
https://github.com/jupyter/docker-stacks.git
synced 2025-10-07 18:14:05 +00:00

* Add cuda12 variant for tensorflow-notebook * Reduce size of CPU version of tensorflow-notebook * Try to fix tests * Update docs/using/selecting.md Co-authored-by: Ayaz Salikhov <mathbunnyru@users.noreply.github.com> * Update images/tensorflow-notebook/cuda12/Dockerfile Co-authored-by: Ayaz Salikhov <mathbunnyru@users.noreply.github.com> * Update tests/docker-stacks-foundation/test_packages.py Co-authored-by: Ayaz Salikhov <mathbunnyru@users.noreply.github.com> * Remove obsolete XLA_FLAGS env var * Install CUDA and cuDNN using pip instead of mamba * Fix pre-commit shell checks * Change tensorflow variant name from cuda12 to cuda * Update selecting.md * Update selecting.md --------- Co-authored-by: Ayaz Salikhov <mathbunnyru@users.noreply.github.com>