출처들

Ubuntu 20.04 (Mint 20.1) CUDA 11.1, cuDNN 8.0.5, Tensorflow 2.4

Tensorflow doesn't seem to see my gpu

개발 환경

우분투 20.04에 드라이버, CUDA, cuDNN 설치하는 과정을 서술 할 거임..

OS 환경

<laptop> CPU : intel i9-12900H GPU : RTX 3070ti

tensorflow 버전에 맞춰서 깔아줄 것임.

<aside> 💡 tensorflow, CUDA, cuDNN 버전 호환 확인은 여기 링크

Windows:

https://www.tensorflow.org/install/source_windows#tested_build_configurations

Mac / Linux:

https://www.tensorflow.org/install/source#tested_build_configurations

</aside>

우리는 2.12.0 버전으로 깔 것이며, linux의 경우에는 tensorflow-gpu를 따로 설치하지 않는다. (2 버전부터는 통합된 듯)

아래 그림처럼 CUDA는 11.8, cuDNN은 8.6 버전으로 설치할 예정임

Untitled

nvidia 드라이버, cuda 삭제