-
https://medium.com/@mccann.matt/compiling-tensorflow-with-cuda-3-0-support-42d8fe0bf3b5
-
Install drivers
-
Install CUDA
-
Install CudNN
-
Install prerequisities (see https://www.tensorflow.org/install/source)
-
Install NCCL
-
Checkout tensorflow
-
Checkout branch
-
Install bazelisk:
- https://github.com/bazelbuild/bazelisk
- Copy to /usr/local/bin
-
./configure
-
bazel build --config=opt --config=cuda --config=numa --config=v2 //tensorflow/tools/pip_package:build_pip_package
-
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
-
sudo pip install /tmp/tensorflow_pkg/tensorflow-2.2.0-cp36-cp36m-linux_x86_64.whl
-
(Disable XLA?)
Created
May 16, 2020 06:24
-
-
Save Pibben/e242608b828df59593798e2720d33e5e to your computer and use it in GitHub Desktop.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment