ModuleNotFoundError: No module named 'nvidia'

Issue

윈도우에서 nvidia.cudnn import 시 또는 TensorFlow 프레임웍 사용 소스 빌드 시 아래와 같은 에러가 발생하는 경우

Traceback (most recent call last):
  File "<string>", line 1, in <module>
ModuleNotFoundError: No module named 'nvidia'

Solution

Windows OS에서는 TensorFlow 2.10까지만 GPU를 지원합니다.

• https://www.tensorflow.org/install/pip

따라서 2.10 이하버전을 설치하거나, 2.11 이상 버전을 사용하기 위해서는 WSL2에 TensorFlow를 설치하거나, tensorflow-gpu를 설치하여 선택적으로 TensorFlow-DirectML-Plugin를 사용해야합니다.

References

• https://stackoverflow.com/questions/75907635/no-matching-distribution-found-in-the-installation-of-the-cudnn-for-tensorflow-v
• https://www.tensorflow.org/install/pip