RE-OWOD/dev/packaging
RE-OWOD f4220c0e51
Add files via upload
2022-01-04 13:17:03 +08:00
..
README.md Add files via upload 2022-01-04 13:17:03 +08:00
build_all_wheels.sh Add files via upload 2022-01-04 13:17:03 +08:00
build_wheel.sh Add files via upload 2022-01-04 13:17:03 +08:00
gen_install_table.py Add files via upload 2022-01-04 13:17:03 +08:00
gen_wheel_index.sh Add files via upload 2022-01-04 13:17:03 +08:00
pkg_helpers.bash Add files via upload 2022-01-04 13:17:03 +08:00

README.md

To build a cu101 wheel for release:

$ nvidia-docker run -it --storage-opt "size=20GB" --name pt  pytorch/manylinux-cuda101
# inside the container:
# git clone https://github.com/facebookresearch/detectron2/
# cd detectron2
# export CU_VERSION=cu101 D2_VERSION_SUFFIX= PYTHON_VERSION=3.7 PYTORCH_VERSION=1.4
# ./dev/packaging/build_wheel.sh

To build all wheels for CUDA {9.2,10.0,10.1} x Python {3.6,3.7,3.8}:

./dev/packaging/build_all_wheels.sh
./dev/packaging/gen_wheel_index.sh /path/to/wheels