• Nikita Shulga's avatar
    Enable Linux wheel/conda GPU package builds (#1730) · 0f844305
    Nikita Shulga authored
    * Remove some obsolete conditions about CUDA-10.0 from `pkg_helpers.bash`.
    * Use `USE_CUDA` instead of `FORCE_CUDA` in `pkg_helpers.bash`
    * Do not define `NO_CUDA_PACKAGE` in build_wheel.sh and build_cuda.sh
    * Add conda-forge for Win cuda-11.1 builds
    * Pass USE_CUDA / TORCH_CUDA_ARCH_LIST to conda build
    * Add selected CUDA to path
    * Don't define USE_CUDA for ROCM
    
    TODO: Fix Windows CUDA builds
    0f844305
pkg_helpers.bash 10 KB