Add CUDA_HOME env var to permit overriding the CUDA install location

This commit is contained in:
Michael Vines
2018-11-10 18:03:38 -08:00
parent b670b9bcde
commit cf6f344ccc
4 changed files with 18 additions and 8 deletions

View File

@ -83,8 +83,10 @@ local|tar)
export RUST_LOG
export SOLANA_DEFAULT_METRICS_RATE=1
: "${CUDA_HOME:=/usr/local/cuda}"
./fetch-perf-libs.sh
export LD_LIBRARY_PATH="$PWD/target/perf-libs:/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
export LD_LIBRARY_PATH="$PWD/target/perf-libs:$CUDA_HOME/lib64:$LD_LIBRARY_PATH"
echo "LD_LIBRARY_PATH=$LD_LIBRARY_PATH"
scripts/oom-monitor.sh > oom-monitor.log 2>&1 &