search

flatironinstitute / nifty-ls

A fast Lomb-Scargle periodogram. It's nifty, and uses a NUFFT!
Apache License 2.0
32 stars 2 forks source link

ci: switch Docker image to `nvidia/cuda:12.6.0-devel-rockylinux9` #35

Closed lgarrison closed 2 months ago

lgarrison commented 2 months ago

cupy currently requires development headers that aren't in the "runtime" CUDA Docker images (https://github.com/cupy/cupy/issues/8466). It seems like eventually the plan is to support CUDA from pip (https://github.com/cupy/cupy/issues/8013), and indeed the headers are already available from the nvidia-cuda-runtime-cu12 on PyPI, but let's not mix and match CUDA sources unless we need to.

(Also, let's use Rocky instead of Ubuntu, since most users will probably be on a GPU cluster running a RHEL variant.)

lgarrison commented 2 months ago

Also, the Jenkins tests weren't running properly on PRs before, but they should be now.