diff options
Diffstat (limited to 'setup_environment')
| -rw-r--r-- | setup_environment | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/setup_environment b/setup_environment index 871bb59..3428101 100644 --- a/setup_environment +++ b/setup_environment @@ -44,7 +44,8 @@ if [ $? = 1 ]; then return 1; fi -CC_VERSION=`gcc --version | head -1 | awk '{for(i=1;i<=NF;i++){ if(match($i,/^[0-9]\.[0-9]\.[0-9]$/)) {print $i; exit 0}}}'` +CC_VERSION=$(gcc --version | head -1 | awk '{for(i=1;i<=NF;i++){ if(match($i,/^[0-9]+\.[0-9]+\.[0-9]+$/)) {print $i; exit 0}}}') + CUDA_VERSION_STRING=`$CUDA_INSTALL_PATH/bin/nvcc --version | awk '/release/ {print $5;}' | sed 's/,//'`; export CUDA_VERSION_NUMBER=`echo $CUDA_VERSION_STRING | sed 's/\./ /' | awk '{printf("%02u%02u", 10*int($1), 10*$2);}'` |