hazza/docker-stacks
1
0
Fork 0
mirror of https://github.com/jupyter/docker-stacks.git synced 2025-10-07 18:14:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add cuda12 variant of tensorflow-notebook (#2100)

Browse Source 
* 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>
This commit is contained in:
ChristofKaufmann
2024-03-26 02:12:59 +01:00
committed by GitHub
parent dd06b93abc
commit b9553a8e5d
6 changed files with 61 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tagging/taggers.py
View File

@@ -98,7 +98,10 @@ class RVersionTagger(TaggerInterface):
class TensorflowVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
return "tensorflow-" + _get_pip_package_version(container, "tensorflow")
try:
return "tensorflow-" + _get_pip_package_version(container, "tensorflow")
except AssertionError:
return "tensorflow-" + _get_pip_package_version(container, "tensorflow-cpu")
class PytorchVersionTagger(TaggerInterface):