https://github.com/ashishpatel26/tpu_tf2

TPU use in single line in colab using tf2 package.
https://github.com/ashishpatel26/tpu_tf2

colab-notebook colaboratory deep-learning tensorflow2 tpu

Last synced: about 1 month ago
JSON representation

TPU use in single line in colab using tf2 package.

• Host: GitHub
• URL: https://github.com/ashishpatel26/tpu_tf2
• Owner: ashishpatel26
• License: mit
• Created: 2021-12-15T06:30:07.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2021-12-15T08:28:52.000Z (over 3 years ago)
• Last Synced: 2025-04-25T09:57:25.566Z (2 months ago)
• Topics: colab-notebook, colaboratory, deep-learning, tensorflow2, tpu
• Language: Jupyter Notebook
• Homepage: https://pypi.org/project/tpu-tf2/
• Size: 5.13 MB
• Stars: 11
• Watchers: 1
• Forks: 3
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# TPU usage in colab with single line of code

## Before installation Make sure your TPU is on

![](https://camo.githubusercontent.com/dd817760c179975842deecf76ca0f31f4ac5a0dfddd6d57a393c1e8ac2156a58/68747470733a2f2f692e737461636b2e696d6775722e636f6d2f41444e666f2e706e67)

## Installation

To install the package from the PyPi repository you can execute the following

command:

```sh

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:

```bash

git clone https://github.com/ashishpatel26/regressionmetrics.git

cd tpu_tf2

pip install .

```

## Usage

> usage in colab : [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/ashishpatel26/tpu_tf2/blob/main/demo_colab.ipynb)

```python

from tpu_tf2.tpu import strategy

strategy.num_replicas_in_sync

> output : 8

```

![](https://github.com/ashishpatel26/tpu_tf2/blob/main/tpu_tf2.gif?raw=true)

💁 Thanks for reading and forking.

---