RuntimeError: "upsample_bilinear2d_out_frame" not implemented for 'BFloat16'

[zhouyifan17]

When I am running os. system (cmd) using GPU, the following error message appears:
RuntimeError: "upsample_bilinear2d_out_frame" not implemented for 'BFloat16'

May I ask how to solve this problem?

[wanghaoyucn]

Hi, please upgrade your torch to 2.1.0 or higher. In earlier versions, upsample doesn't support data type BFloat16

[IGR1chard]

Hi, please upgrade your torch to 2.1.0 or higher. In earlier versions, upsample doesn't support data type BFloat16

hi, i still got this problem when using torch 2.4.1, any suggestions?