also catch errors during installation in case `CUDAExtension` cannot be loaded (#175)
Previously we only catch build errors in `BuildExtension` in https://github.com/facebookresearch/segment-anything-2/pull/155. However, in some cases, the `CUDAExtension` instance might not load. So in this PR, we also catch such errors for `CUDAExtension`.
R
Ronghang Hu committed
6186d1529a9c26f7b6e658f3e704d4bee386d9ba
Parent: 6ecb5ff
Committed by GitHub <noreply@github.com>
on 8/7/2024, 7:26:11 PM