1
运行时
提问帖
有回复
新人贴
RuntimeError: cuFFT error: HIPFFT_INTERNAL_ERROR
2026-08-17 09:15:47
123
求问:想在BW1000上训练模型,但是显示了这个FFT错误,相关环境在最后 ----------------------------------------------------------------- Traceback (most recent call last): File "/home/ubuntu/CellWarper\_HaiGuang/src/CellWarper/train.py", line 2197, in train(config\_dir, args, rank, world\_size, local\_rank, logger) File "/home/ubuntu/CellWarper\_HaiGuang/src/CellWarper/train.py", line 1756, in train loss\_dict = criterion( ^^^^^^^^^^ File "/home/ubuntu/miniconda3/envs/dcu/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1751, in \_wrapped\_call\_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/miniconda3/envs/dcu/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/CellWarper_HaiGuang/src/CellWarper/train.py", line 858, in forward loss_tildeq = tildeq_loss_single_var(pred, target, alpha=self.tildeq_alpha, gamma=self.tildeq_gamma) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/CellWarper_HaiGuang/src/CellWarper/utils/tildeq_loss.py", line 189, in tildeq_loss_single_var l_amp = tildeq_amp_loss(outputs, targets) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/CellWarper_HaiGuang/src/CellWarper/utils/tildeq_loss.py", line 52, in tildeq_amp_loss out_fourier = torch.fft.fft(outputs_f32, fft_size, dim=-1) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: cuFFT error: HIPFFT_INTERNAL_ERROR 相关环境: 显卡:BW1000 驱动:rock-6.3.31-V1.5.3.run dtk:DTK-26.04-Ubuntu22.04-x86_64.tar.gz 操作系统:Ubuntu Server 24.04 LTS 64位 UEFI版 HCC miniconda:Miniconda3-py312_26.5.3-2-Linux-x86_64.sh torch:torch-2.7.1+das.opt1.dtk2604-cp312-cp312-manylinux_2_28_x86_64.whl ``` torchaudio-2.7.1+das.opt1.dtk2604.torch271-cp312-cp312-manylinux_2_28_x86_64.whl ``` ``` torchvision-0.22.0+das.opt1.dtk2604.torch271-cp312-cp312-manylinux_2_28_x86_64.whl ``` 请问下这个错误是什么原因呢
全部评论(3)
只看楼主
L
lim
2楼
export ROCFFT_LAYER=3 打印下FFT日志
8/19
L
lim
1楼
请问您跑的模型是哪个 能提供下复现步骤吗
8/19
。
。。。
回复 @lim
之前联系海光工作人员尝试复现bug,没有复现成功,但莫名其妙突然可以正常训练了,目前还没有搞清楚怎么回事
8/20
没有更多内容了
--
楼主
Level.6
0 帖子
楼主热帖
暂无数据