The goal for this PR is to patch Tensorflow r2.0 release, so it would fully enable ROCm non-xla path support. Most of the PRs been cherry-picked in this patch have already been upstreamed in the upstream master branch. The following were all the related commits been cherry-picked: Commits on Aug 20, 2019 deven-amd and sunway513 adding/updating ROCm support in the ci_build scripts d5a0eee deven-amd and sunway513 updating Dockerfile.rocm to pick a specific version of the rocm libra… … e335575 deven-amd and sunway513 adding a script for testing the ROCm Community Supported Build ae83a20 Commits on Aug 22, 2019 deven-amd and sunway513 Resolve merge conflicts for PR #31393 73ff708 deven-amd and sunway513 The following PR/commit breaks the --config=rocm build … 614bdb5 deven-amd and sunway513 updating testcases to work correctly with ROCm 1685240 jeffdaily and sunway513 improve concurrency between compute and nccl streams … 3fbb049 whchung and sunway513 [ROCm] enable roll op on ROCm. 1d5f440 whchung and sunway513 [ROCm] enable InTopK op on ROCm. 941f713 deven-amd and sunway513 updating README.md with information on ROCm Community Supported Builds 73ce64e Commits on Aug 25, 2019 houtoms and sunway513 fixed potential rocm breaks from use_padded_io 0832b33 deven-amd and sunway513 adding no_rocm tag on unit-tests that check features that are current… … 7aed626 deven-amd and sunway513 Adding ROCm support for reduction ops 82bd216 sunway513 Fix ROCm path build error in rocm_dnn.h 5dba305 Commits on Aug 27, 2019 deven-amd fixing test failures by skipping parts that functionality not yet sup… … be6378c sunway513 Merge pull request #616 from ROCmSoftwarePlatform/r2.0-rocm-upstream-… … d98a943 sunway513 Add no_rocm tag to //tensorflow/python:stateful_random_ops_test_gpu d05a47f Commits on Sep 04, 2019 sunway513 Merge branch 'r2.0-rocm-upstream' of https://github.com/ROCmSoftwareP… … b1148e4 Commits on Sep 06, 2019 deven-amd and sunway513 adding ROCm support in the build_pip_package script b908324
12 KiB

Documentation |
---|
TensorFlow is an open source software library for numerical computation using data flow graphs. The graph nodes represent mathematical operations, while the graph edges represent the multidimensional data arrays (tensors) that flow between them. This flexible architecture enables you to deploy computation to one or more CPUs or GPUs in a desktop, server, or mobile device without rewriting code. TensorFlow also includes TensorBoard, a data visualization toolkit.
TensorFlow was originally developed by researchers and engineers working on the Google Brain team within Google's Machine Intelligence Research organization for the purposes of conducting machine learning and deep neural networks research. The system is general enough to be applicable in a wide variety of other domains, as well.
TensorFlow provides stable Python and C APIs as well as non-guaranteed backwards compatible API's for C++, Go, Java, JavaScript, and Swift.
Keep up to date with release announcements and security updates by subscribing to announce@tensorflow.org.
Installation
To install the current release for CPU-only:
pip install tensorflow
Use the GPU package for CUDA-enabled GPU cards:
pip install tensorflow-gpu
See Installing TensorFlow for detailed instructions, and how to build from source.
People who are a little more adventurous can also try our nightly binaries:
Nightly pip packages * We are pleased to announce that TensorFlow now offers
nightly pip packages under the
tf-nightly and
tf-nightly-gpu project on PyPi.
Simply run pip install tf-nightly
or pip install tf-nightly-gpu
in a clean
environment to install the nightly TensorFlow build. We support CPU and GPU
packages on Linux, Mac, and Windows.
Try your first TensorFlow program
$ python
>>> import tensorflow as tf
>>> tf.enable_eager_execution()
>>> tf.add(1, 2).numpy()
3
>>> hello = tf.constant('Hello, TensorFlow!')
>>> hello.numpy()
'Hello, TensorFlow!'
Learn more examples about how to do specific tasks in TensorFlow at the tutorials page of tensorflow.org.
Contribution guidelines
If you want to contribute to TensorFlow, be sure to review the contribution guidelines. This project adheres to TensorFlow's code of conduct. By participating, you are expected to uphold this code.
We use GitHub issues for tracking requests and bugs, please see TensorFlow Discuss for general questions and discussion, and please direct specific questions to Stack Overflow.
The TensorFlow project strives to abide by generally accepted best practices in open-source software development:
Continuous build status
Official Builds
Build Type | Status | Artifacts |
---|---|---|
Linux CPU | pypi | |
Linux GPU | pypi | |
Linux XLA | TBA | |
MacOS | pypi | |
Windows CPU | pypi | |
Windows GPU | pypi | |
Android | ||
Raspberry Pi 0 and 1 | Py2 Py3 | |
Raspberry Pi 2 and 3 | Py2 Py3 |
Community Supported Builds
Build Type | Status | Artifacts |
---|---|---|
Linux AMD ROCm GPU Nightly | Nightly | |
Linux AMD ROCm GPU Stable Release | Release | |
Linux s390x Nightly | Nightly | |
Linux s390x CPU Stable Release | Release | |
Linux ppc64le CPU Nightly | Nightly | |
Linux ppc64le CPU Stable Release | Release | |
Linux ppc64le GPU Nightly | Nightly | |
Linux ppc64le GPU Stable Release | Release | |
Linux CPU with Intel® MKL-DNN Nightly | Nightly | |
Linux CPU with Intel® MKL-DNN Supports Python 2.7, 3.4, 3.5, and 3.6 |
1.13.1 pypi | |
Red Hat® Enterprise Linux® 7.6 CPU & GPU Python 2.7, 3.6 |
1.13.1 pypi |
For more information
- TensorFlow Website
- TensorFlow Tutorials
- TensorFlow Model Zoo
- TensorFlow Twitter
- TensorFlow Blog
- TensorFlow Course at Stanford
- TensorFlow Roadmap
- TensorFlow White Papers
- TensorFlow YouTube Channel
- TensorFlow Visualization Toolkit
Learn more about the TensorFlow community at the community page of tensorflow.org for a few ways to participate.