[WSL] CUDA Version Downgrade

wsl을 통해 코드를 실행하려니, 버전이 안 맞는 에러가 발생했다.

TensorFlow가 GPU를 사용하려고 시도했으나, CUDA 또는 cuDNN 라이브러리 초기화에 실패했습니다.

tensorflow.python.framework.errors_impl.FailedPreconditionError: Graph execution error와 DNN library initialization failed

---

나의 CUDA Version은 12.8이다.

 

나는 파이썬 3.10을 사용하기에, 

CUDA는 11.2가, 

tensorflow는 2.8.0 or 2.9.0 or 2.10.0이 필요하다...

https://www.tensorflow.org/install/source_windows?hl=ko#tested_build_configurations

Windows의 소스에서 빌드,Windows의 소스에서 빌드  |  TensorFlow

---

1-1. CUDA 변경 : CUDA 12.8 대신 CUDA 11.2로 설치

터미널에서 다음 명령어로 현재 설치된 CUDA 및 cuDNN을 제거하세요 ( NVIDIA 드라이버는 유지합니다. ) :

sudo apt purge nvidia-cuda-toolkit
sudo apt purge libcudnn*
sudo apt autoremove
sudo apt clean

 

1-2. CUDA 11.2 설치 파일 다운로드:
NVIDIA 공식 페이지에서 CUDA 11.2 다운로드: CUDA Toolkit Archive

CUDA Version이 변경이 안 되었다면... 아래 접은 글 참조...

CMD(명령 프롬프트) 기본 모드에서는 cuda 11.2가 되는데,

CMD(명령 프롬프트) 관리자 모드에서는 cuda 12.8이 여전히 나온다.

 

👉 WSL에 CUDA 11.2를 사용하고 싶다면

1. CUDA 11.2 패키지 설치:
2.  
   sudo apt update sudo apt install cuda-11-2
3. 환경 변수 설정:
4.  
   echo 'export PATH=/usr/local/cuda-11.2/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.2/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc

안됨. 에러 발생.

 

마지막에 E: Unable to locate package cuda-11-2 오류는 해당 패키지를 찾을 수 없다는 의미입니다. 이는 CUDA 저장소가 추가되지 않았거나 패키지 이름이 정확하지 않기 때문입니다.

 

✅ 해결 방법:

1. NVIDIA CUDA 저장소 추가

먼저, WSL 환경에서 CUDA 11.2를 설치하려면 NVIDIA의 공식 CUDA 저장소를 추가해야 합니다.

# NVIDIA GPG 키 추가
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub

# NVIDIA CUDA 저장소 추가 (Ubuntu 22.04 기준)
sudo sh -c 'echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /" > /etc/apt/sources.list.d/cuda.list'

# 패키지 목록 업데이트
sudo apt update

 

아ㅏㅏㅏㅏ 그래도 안 된다,,,

그냥 가상환경 새로 만들자,,,

 

2. TensorFlow 변경:Python 3.10에 맞는 TensorFlow GPU 버전 

• tensorflow-gpu==2.10.0 (가장 최신 및 안정적)
• 또는 tensorflow-gpu==2.9.0
• 또는 tensorflow-gpu==2.8.0

# PyTorch 버전 조정:torch==2.5.1와 torchvision==0.20.1으로 다운그레이드
pip uninstall torch torchvision -y
pip install torch==2.5.1 torchvision==0.20.1 --index-url https://download.pytorch.org/whl/cu118


# CUDA 및 cuDNN 변경 후
pip uninstall tensorflow tensorflow-io-gcs-filesystem -y
pip install tensorflow-gpu==2.10.0