求问:想在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
```
请问下这个错误是什么原因呢