build xformers v16 (colab bc08bbc)

Versão padrão

compilada usando o Google colab e o seguinte comando

!git clone https://github.com/facebookresearch/xformers.git; cd xformers; git submodule update --init --recursive; pip install -r ./requirements.txt; python setup.py build develop sdist bdist_wheel --universal

Versão cuda (6.0, 7.0, 7.5, 8.0 e 8.6)

compilada usando o Google colab e o seguinte comando

!git clone https://github.com/facebookresearch/xformers.git; cd xformers; git submodule update --init --recursive; pip install -r ./requirements.txt; TORCH_CUDA_ARCH_LIST="6.0;7.0;7.5;8.0;8.6" python setup.py build develop sdist bdist_wheel --universal

Para mais informações veja o repositório oficial aqui

build xformers (colab 07ba3f3)

Xformers pré-compilado (07ba3f3)

Para fazer seu próprio build rode isso no google colab:

!git clone https://github.com/facebookresearch/xformers.git; cd xformers; git submodule update --init --recursive; pip install -r ./requirements.txt; python setup.py build develop