Implementation of Mesh-TensorFlow
A detailed user manual for mesh-tensorflow
can be found here: https://github.com/tensorflow/mesh.
To use mesh-tensorflow
, you first need to install it, as follows:
# installation #first step pip3 install tensorflow # GPU support pip3 install tensorflow-gpu #second step pip3 install mesh-tensorflow
After that, you can use mesh-tensorflow
directly for distributed tensor operations.