Try hermetic CUDA builds (matching system installed CUDA and cuDNN version)

NVIDIA / JAX-Toolbox

JAX-Toolbox

Apache License 2.0

234 stars 44 forks source link

Try hermetic CUDA builds (matching system installed CUDA and cuDNN version) #999

Closed gpupuck closed 1 month ago

gpupuck commented 1 month ago

This is only a temporary WAR to unblock the JAX and DLFW image builds. The resulting image sizes do not change, compared to the ones we built using system CUDA libs.