This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
### Do not run this script directly | |
# This script is intended to go along with | |
# BuildTensorflow.sh and remove some of the | |
# more complicated functions that result | |
# in a simple setting of environment variables | |
whereami=`pwd` | |
cuda_location=$(dirname $(dirname $(which nvcc))) | |
# Get at the deviceQuery sample code to extract useful information |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/bin/bash | |
#Jason T. 2-6-2018 | |
# Check specifically for the run command | |
if [[ $# -ge 2 && $1 == "run" ]]; then | |
# Tell docker to share the following folders with the base system | |
# This allows the docker containers to find CUDA, cuDNN, TensorRT | |
LIB_MAPS="/usr/lib/aarch64-linux-gnu \ | |
/usr/local/cuda \ | |
/usr/local/cuda/lib64" |