23 lines
852 B
Bash
23 lines
852 B
Bash
git clone --recursive https://github.com/tinygrad/tinygrad
|
|
cd tinygrad/
|
|
python3 -m venv venv
|
|
source venv/bin/activate
|
|
pip install -U setuptools pip wheel
|
|
# Tinygrad from git repo
|
|
pip install -e .
|
|
# or from pypi
|
|
# pip install tinygrad
|
|
|
|
# To run the various examples and benchmarks:
|
|
# Torch's ROCm version of Torch:
|
|
# pip install --pre torch torchaudio torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.0
|
|
# Or AMD ROCm version of Torch:
|
|
pip install torch==2.1.1 torchvision==0.16.1 -f https://repo.radeon.com/rocm/manylinux/rocm-rel-6.0
|
|
|
|
pip install librosa nltk phonemizer protobuf pyyaml \
|
|
sentencepiece tiktoken unidecode gymnasium pytest hypothesis \
|
|
pillow opencv-python tensorflow ultralytics onnx pygame ctypeslib2 \
|
|
tf2onnx lm_eval onnxruntime pydot tensorflow_addons
|
|
# If portaudio.h is available
|
|
pip install pyaudio
|