mirror of
https://github.com/jupyter/docker-stacks.git
synced 2025-10-16 22:42:57 +00:00
3fef15446f
* Put nvidia-smi on path for cuda image variants * Improve comments style --------- Co-authored-by: Ayaz Salikhov <mathbunnyru@users.noreply.github.com>
10 lines
459 B
Bash
10 lines
459 B
Bash
#!/bin/bash
|
|
# Copyright (c) Jupyter Development Team.
|
|
# Distributed under the terms of the Modified BSD License.
|
|
|
|
# This adds NVIDIA Python package 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
|