From f9d3c50fa29a960a2de088449bcb34f046c09685 Mon Sep 17 00:00:00 2001 From: Marcel Ochsendorf Date: Wed, 9 Aug 2017 03:23:27 +0200 Subject: [PATCH 1/2] Update buildTensorFlow.sh --- buildTensorFlow.sh | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/buildTensorFlow.sh b/buildTensorFlow.sh index ec39453..ff6809f 100755 --- a/buildTensorFlow.sh +++ b/buildTensorFlow.sh @@ -1,4 +1,13 @@ #!/bin/bash + + +#FIRST CREATE A SWAP FILE OF 8 GB +fallocate -l 8G swapfile +chmod 600 swapfile +mkswap swapfile +swapon swapfile +swapon -s + # NVIDIA Jetson TX1 # TensorFlow Installation # Export TensorFlow GPU environment variables From c54090819005d718ac3e06ff4ba3a79a3b6e47fe Mon Sep 17 00:00:00 2001 From: Marcel Ochsendorf Date: Wed, 9 Aug 2017 21:37:01 +0200 Subject: [PATCH 2/2] Update buildTensorFlow.sh --- buildTensorFlow.sh | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/buildTensorFlow.sh b/buildTensorFlow.sh index ff6809f..9fd76f8 100755 --- a/buildTensorFlow.sh +++ b/buildTensorFlow.sh @@ -2,10 +2,10 @@ #FIRST CREATE A SWAP FILE OF 8 GB -fallocate -l 8G swapfile -chmod 600 swapfile -mkswap swapfile -swapon swapfile +fallocate -l 8G /swapfile +chmod 600 /swapfile +mkswap /swapfile +swapon /swapfile swapon -s # NVIDIA Jetson TX1 @@ -15,7 +15,7 @@ swapon -s export TF_NEED_CUDA=1 export TF_CUDA_VERSION=8.0 export CUDA_TOOLKIT_PATH=/usr/local/cuda -export TF_CUDNN_VERSION=5.1.10 +export TF_CUDNN_VERSION=6.0.21 export CUDNN_INSTALL_PATH=/usr/lib/aarch64-linux-gnu/ export TF_CUDA_COMPUTE_CAPABILITIES=6.2 @@ -23,3 +23,7 @@ export TF_CUDA_COMPUTE_CAPABILITIES=6.2 cd $HOME/tensorflow bazel build -c opt --local_resources 3072,4.0,1.0 --verbose_failures --config=cuda //tensorflow/tools/pip_package:build_pip_package +#Remove Swapfile +swapoff /swapfile +swapoff -a -v +rm /swapfile