xformer build environment based on CUDA 11.3.1
50
Build process of
xformersis referenced from 리눅스 xformers 빌드 방법 (GPU 불필요) (작성 중)
xformers-builder Docker ImageDOCKER_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 .
xformers-builder Containergit submodule update --init --recursiveMAX_JOBS based on your needdocker 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 directoryAn image and an entrypoint is based on CUDA 11.3 You can modify it based on your need and for use in custom environment
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