mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
bda688c186
Signed-off-by: Edward Z. Yang <ezyang@fb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/88501 Approved by: https://github.com/wconstab
41 lines
2.4 KiB
Plaintext
41 lines
2.4 KiB
Plaintext
# Makefile for the dashboard setup
|
|
TRITON_VERSION ?= $(shell cat ../../.github/ci_commit_pins/triton.txt)
|
|
PIP ?= python -m pip
|
|
|
|
clone-deps:
|
|
(cd ../../.. \
|
|
&& (test -e torchvision || git clone --recursive https://github.com/pytorch/vision torchvision) \
|
|
&& (test -e torchdata || git clone --recursive https://github.com/pytorch/data.git torchdata) \
|
|
&& (test -e torchtext || git clone --recursive https://github.com/pytorch/text torchtext) \
|
|
&& (test -e torchaudio || git clone --recursive https://github.com/pytorch/audio torchaudio) \
|
|
&& (test -e detectron2 || git clone --recursive https://github.com/facebookresearch/detectron2) \
|
|
&& (test -e torchbenchmark || git clone --recursive https://github.com/pytorch/benchmark torchbenchmark) \
|
|
&& (test -e triton || git clone --recursive https://github.com/openai/triton.git) \
|
|
)
|
|
|
|
pull-deps: clone-deps
|
|
echo $(TRITON_VERSION)
|
|
(cd ../../../torchvision && git pull && git submodule update --init --recursive)
|
|
(cd ../../../torchdata && git pull && git submodule update --init --recursive)
|
|
(cd ../../../torchtext && git pull && git submodule update --init --recursive)
|
|
(cd ../../../torchaudio && git pull && git submodule update --init --recursive)
|
|
(cd ../../../detectron2 && git pull && git submodule update --init --recursive)
|
|
(cd ../../../torchbenchmark && git pull && git submodule update --init --recursive)
|
|
(cd ../../../triton && git checkout master && git pull && git checkout $(TRITON_VERSION) && git submodule update --init --recursive)
|
|
|
|
build-deps: clone-deps
|
|
# conda env remove --name torchdynamo
|
|
# conda create --name torchdynamo -y python=3.8
|
|
# conda activate torchdynamo
|
|
conda install -y astunparse numpy scipy ninja pyyaml mkl mkl-include setuptools cmake \
|
|
cffi typing_extensions future six requests dataclasses protobuf numba cython scikit-learn
|
|
conda install -y -c pytorch magma-cuda116
|
|
conda install -y -c conda-forge librosa
|
|
(cd ../../../torchvision && python setup.py clean && python setup.py develop)
|
|
(cd ../../../torchdata && python setup.py install)
|
|
(cd ../../../torchtext && python setup.py clean && python setup.py develop)
|
|
(cd ../../../torchaudio && python setup.py clean && python setup.py develop)
|
|
(cd ../../../detectron2 && python setup.py clean && python setup.py develop)
|
|
(cd ../../../torchbenchmark && python install.py --continue_on_fail)
|
|
(cd ../../../triton/python && python setup.py clean && python setup.py develop)
|