[Linux] Ubuntu 22.04 RTX 4090 RuntimeError: cuFFT error 해결

환경세팅을 마치고 잘 돌아가는지 학습을 돌렸는데,

스펙트로그램 변환에서 stft를 거치다가

RuntimeError: cuFFT error: CUFFT_INTERNAL_ERROR 가 떴다

 

와이이

 

구글링 해보니 4090에서 종종 발생하는 에러이며 특히 CUDA 11.7에서 이슈가 발생한다는 것을 확인

CUDA 11.8로 실행 시 문제가 없다함

기존 CUDA 11.7에서 11.8로 설치하고 진행

0. 기존 파이토치 삭제하기

 

나의 경우 conda로 설치하였기 때문에

아래 명령어 입력

 

conda uninstall pytorch

 

pip로 설치 시 아래 명령어 입력

 

pip uninstall torch
pip uninstall torch # 꼭 두 번 실행

 

1. CUDA 11.8에 맞는 파이토치 재설치

여기 에서 필요에 맞는 옵션 클릭하거나

 

 

conda로 CUDA 11.8에 맞는 Pytorch 2.0.0 버전을 설치하려면 아래 명령어 입력

 

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

 

2. Pytorch 와 CUDA 버전 확인

아래 명령어로 확인 가능하다

 

# 파이토치 버전
print(torch.__version__)

# CUDA 사용 여부
torch.cuda.is_available()

# 파이토치 CUDA 버전
torch.version.cuda

 

 

3. 에러 해결 확인

다시 학습을 실행시켜보니 에러 없이 잘 돌아간다

 

 

끝!