Sign inSign up

kestr3l/xformer-builder

By kestr3l

•Updated almost 4 years ago

xformer build environment based on CUDA 11.3.1

Image
0

50

kestr3l/xformer-builder repository overview

⁠Xformers-Builder

⁠0 Prequisites

⁠1 xformers-builder (Optional)

Build process of xformers is referenced from 리눅스 xformers 빌드 방법 (GPU 불필요) (작성 중)⁠

⁠1.1 Build xformers-builder Docker Image
DOCKER_BUILDKIT=1 docker build --no-cache \
--build-arg BASEIMAGE=nvidia/cuda \
--build-arg BASETAG=11.3.1-devel-ubuntu20.04 \
-t kestr3l/xformer-builder:cu113-1.0.0 \
-f Dockerfile .
⁠1.2 Build xfomers using xformers-builder Container
  • Clone facebookresearch/xformers⁠ and update submodules
    • git submodule update --init --recursive
  • Then mount xformers directory into docker container. Set MAX_JOBS based on your need
docker run -it --rm \
     -e NVIDIA_DISABLE_REQUIRE=1 \
     -e NVIDIA_DRIVER_CAPABILITIES=all \
     -e MAX_JOBS=$(nproc) \
     -v /dir/to/xformers/repo:/root/xformers \
     --gpus all \
     --privileged \
     kestr3l/xformer-builder:cu113-1.0.0
  • .whl file will be generated on xforemrs/dist directory

An image and an entrypoint is based on CUDA 11.3 You can modify it based on your need and for use in custom environment

⁠2 References

  1. facebookreseasrch/xformers⁠
  2. 리눅스 xformers 빌드 방법 (GPU 불필요) (작성 중)⁠

Tag summary

Content type

Image

Digest

sha256:811fbe1d7…

Size

2.2 GB

Last updated

almost 4 years ago

docker pull kestr3l/xformer-builder:cu113-1.0.0