TPU use in single line in colab.
Project description
TPU usage in colab with single line of code
Before installation Make sure your TPU is on
Installation
To install the package from the PyPi repository you can execute the following command:
pip install tpu_tf2
If you prefer, you can clone it and run the setup.py file. Use the following commands to get a copy from GitHub and install all dependencies:
git clone https://github.com/ashishpatel26/regressionmetrics.git
cd tpu_tf2
pip install .
Usage
usage in colab :
from tpu_tf2.tpu import strategy
strategy.num_replicas_in_sync
> output : 8
💁 Thanks for reading and forking.
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
tpu_tf2-1.1.0.tar.gz
(3.0 kB
view hashes)
