mirror of
https://github.com/jupyter/docker-stacks.git
synced 2025-10-17 23:12:56 +00:00
b9553a8e5d
* 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>
10 lines
448 B
Bash
10 lines
448 B
Bash
#!/bin/bash
|
|
# Copyright (c) Jupyter Development Team.
|
|
# Distributed under the terms of the Modified BSD License.
|
|
|
|
# This adds the NVIDIA libraries to the LD_LIBRARY_PATH. Workaround for
|
|
# https://github.com/tensorflow/tensorflow/issues/63362
|
|
NVIDIA_DIR=$(dirname "$(python -c 'import nvidia;print(nvidia.__file__)')")
|
|
LD_LIBRARY_PATH=$(echo "${NVIDIA_DIR}"/*/lib/ | sed -r 's/\s+/:/g')${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
|
|
export LD_LIBRARY_PATH
|