模型对50系显卡适配有问题

官方文档给的是pytorch=2.0.1，最高支持sm_90架构，50系为sm_120架构，会出现识别不到显卡。