Error when running preprocess.py

[PS-AI]

Hi,
When running preprocess.py, I get this error:
RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
Could you guide me on solving this?

This is the stack trace:

Loading hubert for content...
load model(s) from /home/ubuntu/PS/NaturalSpeech2/NS2VC/checkpoint_best_legacy_500.pt
Loaded hubert.
0%| | 0/43950 [00:01<?, ?it/s]
Process Process-1:
Traceback (most recent call last):
File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap
self.run()
File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run
self._target(*self._args, **self._kwargs)
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/preprocess.py", line 73, in process_batch
process_one(filename, hmodel, codec)
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/preprocess.py", line 34, in process_one
c = utils.get_hubert_content(hmodel, wav_16k_tensor=wav16k)
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/utils.py", line 235, in get_hubert_content
logits = hmodel.extract_features(**inputs)
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/venv38/lib/python3.8/site-packages/fairseq/models/hubert/hubert.py", line 535, in extract_features
res = self.forward(
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/venv38/lib/python3.8/site-packages/fairseq/models/hubert/hubert.py", line 437, in forward
features = self.forward_features(source)
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/venv38/lib/python3.8/site-packages/fairseq/models/hubert/hubert.py", line 392, in forward_features
features = self.feature_extractor(source)
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/venv38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/venv38/lib/python3.8/site-packages/fairseq/models/wav2vec/wav2vec2.py", line 895, in forward
x = conv(x)
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/venv38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/venv38/lib/python3.8/site-packages/torch/nn/modules/container.py", line 217, in forward
input = module(input)
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/venv38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/venv38/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 313, in forward
return self._conv_forward(input, self.weight, self.bias)
File "/home/ubuntu/PS/NaturalSpeech2/NS2VC/venv38/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 309, in _conv_forward
return F.conv1d(input, weight, bias, self.stride,
RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR