在RTX5090上的torch 和 torchvision和CUDA的兼容问题

[qj164]

RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions. 报错

[Louym]

Could you try this file? Just replace the original setup.py.