【报错】RuntimeError: HIP error: invalid device function

【报错】

RuntimeError: HIP error: invalid device function

HIP kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.

For debugging consider passing AMD_SERIALIZE_KERNEL=3

Compile with `TORCH_USE_HIP_DSA` to enable device-side assertions.

【分析】

使用的是官方基础镜像rocm/pytorch:latest，GPU的