Skip to content

Update buildTensorFlow.sh #10

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Update buildTensorFlow.sh #10

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
f9d3c50
Update buildTensorFlow.sh
RBEGamer Aug 9, 2017
c540908
Update buildTensorFlow.sh
RBEGamer Aug 9, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 14 additions & 1 deletion buildTensorFlow.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,16 +1,29 @@
#!/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
# WARNING This needs to match setTensorFlowEV.sh settings
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

# Build Tensorflow
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