coolnalu/build.sh

Last active May 28, 2017 18:20

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/coolnalu/51df4edbef38d29316d413554d351262.js"></script>
Save coolnalu/51df4edbef38d29316d413554d351262 to your computer and use it in GitHub Desktop.

Download ZIP

Compiling TensorFlow on Ubuntu 16.04 LTS with Nvidia cuDNN and cuda

Raw

bazel build -c opt --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-mfpmath=both --copt=-msse4.2 --config=cuda -k //tensorflow/tools/pip_package:build_pip_package

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment