Clone Cerebras Model Zoo#
Cerebras’s Model Zoo repository contains examples of common deep learning models that can be trained on the Cerebras Wafer-Scale cluster. A list of supported models can be found here.
Follow the steps on this page to clone the Cerebras Model Zoo repository in your preferred directory.
Clone repository#
Clone the Cerebras repository by using one of the following methods:
git clone https://github.com/Cerebras/modelzoo.git
git checkout tags/Release_2.1.1 -b main
or
git clone -b Release_2.1.1 https://github.com/Cerebras/modelzoo.git
Installation#
Once you have Set up a Cerebras virtual environment, install additional packages to your Python environment to support all the functionalities of the Cerebras Model Zoo.
Install packages available in Cerebras Model Zoo by issuing the following commands:
source venv_cerebras_pt/bin/activate
Note that now you should be in the (venv_cerebras_pt) environment.
pip install --upgrade pip
pip install -r modelzoo/requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu
Note
Refer to Run Cerebras Model Zoo on a GPU if you are running your model on a GPU.