ITADN

使用RTX50系显卡出现CUDA错误

#137Openldxldx123 创建于 2025-02-05
L
ldxldx123commented
NVIDIA GeForce RTX 5080 with CUDA capability sm_120 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 sm_90. If you want to use the NVIDIA GeForce RTX 5080 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/ Traceback (most recent call last): File "train_reflow.py", line 78, in <module> train(args, initial_global_step, model, optimizer, scheduler, vocoder, loader_train, loader_valid) File "F:\So-VITS-SVC\DDSP-barbara-6.1\reflow\solver.py", line 232, in train ddsp_loss, reflow_loss=model(data['units'], data['f0'], data['volume'], data['spk_id'], File "F:\So-VITS-SVC\DDSP-barbara-6.1\env\lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "F:\So-VITS-SVC\DDSP-barbara-6.1\env\lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl return forward_call(*args, **kwargs) File "F:\So-VITS-SVC\DDSP-barbara-6.1\reflow\vocoder.py", line 195, in forward ddsp_wav, hidden = self.ddsp_model(units, f0, volume, spk_id=spk_id, spk_mix_dict=spk_mix_dict, aug_shift=aug_shift, infer=infer) File "F:\So-VITS-SVC\DDSP-barbara-6.1\env\lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "F:\So-VITS-SVC\DDSP-barbara-6.1\env\lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl return forward_call(*args, **kwargs) File "F:\So-VITS-SVC\DDSP-barbara-6.1\ddsp\vocoder.py", line 535, in forward combtooth = self.fast_source_gen(f0_frames) File "F:\So-VITS-SVC\DDSP-barbara-6.1\ddsp\vocoder.py", line 515, in fast_source_gen n = torch.arange(self.block_size, device=f0_frames.device) RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1. Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.
4 条评论