Tensorflow(>=2.15) WSL2 설치 (GPU, CUDA)

 

먼저, nvidia-smi 명령을 통해 TensorFlow에서 요구하는 CUDA 버전보다 같거나 높은지 확인하고 낮을 경우에는 윈도우 상에서 NVIDA 그래픽 드라이버 업데이트를 진행하여야 합니다.

https://www.tensorflow.org/install/source?hl=ko#gpu

https://docs.nvidia.com/deploy/cuda-compatibility/#minor-version-compatibility

 

 

 

버전 확인 후,

python3 -m pip install tensorflow[and-cuda]

실행하여 TensorFlow를 설치합니다.

(TF 2.15 버전부터는 CUDA 라이브러리 등 GPU 지원을 위해 수동으로 설치해야 되었던 항목들이 모두 포함되어 자동으로 설치됩니다.)

https://blog.tensorflow.org/2023/11/whats-new-in-tensorflow-2-15.html

 

설치 이후에는

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

명령을 실행하여 GPU가 정상적으로 잡히고 있는지 확인할 수 있습니다.

https://www.tensorflow.org/install/pip?hl=ko#windows-wsl2