PyTorch框架模型训练迁移时报错“Stream mode cannot be set in current driver version”
2024/02/21
106
问题信息
问题来源 | 产品大类 | 产品子类 | 关键字 |
---|---|---|---|
官方 | 模型训练 | PyTorch | -- |
问题现象描述
- 硬件配置:
- CANN版本:6.3.RC1
- Driver版本:5.1.RC1
- Torch版本:1.8.1
- 问题现象:
在AICC上迁移PyTorch框架模型在NPU上训练时,驱动层有报错“Stream mode cannot be set in current driver version”。
原因分析
c84有一个遇错即停的特性,分别合入了CANN和驱动包,该功能必须两个都是c84以上才会生效,但是torch包把这个功能默认打开了,所以就会有上面的报错,但是不影响业务和功能。
解决措施
升级driver版本为C84版本即可。
本页内容