Skip to content

Instantly share code, notes, and snippets.

View JasonAtNvidia's full-sized avatar

JasonAtNvidia

  • NVIDIA
  • Washington DC
View GitHub Profile
### 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
@JasonAtNvidia
JasonAtNvidia / txdocker
Last active March 8, 2022 08:15
Short Bash script to enable use of the GPU within a docker container running on an NVIDIA Jetson TX2. Place inside /usr/local/bin/, chmod +x txdocker, ensure it is in your system PATH, and use just as you would the docker command.
#!/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"