This repository has been archived by the owner on Oct 22, 2023. It is now read-only.
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