Skip to content

GitLab

"model/vscode:/vscode.git/clone" did not exist on "5e2e0b46b1b4ae7145011642631a7b507a70fb03"
Switch branch/tag
Find file BlameHistoryPermalink
  • 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