[에러기록] CUDA, cuDNN 인식이 안 되는 경우

 

가끔 CUDA, cuDNN을 설치했는데 파이썬 라이브러리에서 인식을 못 하는 경우가 있다.

 

컴퓨터 자체에 CUDA, cuDNN을 설치하는 것과 가상환경에서 CUDA, cuDNN을 설치하는 것에 따라 인식에 문제가 있을 수 있다.

 

컴퓨터에 CUDA,cuDNN을 설치했었으나 여러 개의 CUDA를 설치하는 것이 바람직하지 않다는 피드백을 받았다. 따라서 가상환경을 만들고 그 안에 CUDA toolkit을 설치하는 것이 바람직해보인다.

 

https://stackoverflow.com/questions/73424216/how-to-install-cudnn-and-cuda-toolkit-in-a-python-virtual-environment

How to install cudnn and cuda toolkit in a python virtual environment?

다음 stackoverflow에도 비슷한 맥락으로 가상환경 내에 cuda를 설치하는 방법에 대해 바룬다.

 

 

 

 

1. 가상환경 내에 설치된 CUDA 버전이 적절하지 않은 경우

 

보통은 가상환경을 만들고 그 안에

 

conda install --channel "nvidia/label/cuda-12.6.0" cuda    
pip install "jax[cuda12]==0.4.29"

 

2. cuDNN 버전과 CUDA 버전 호환성이 안 맞는 경우

 

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0