Anaconda에서 GPU 사용 가능한 Tensorflow 설치하기

이 문제 때문에 서너시간은 족히 쓴 것 같네요. 거두절미 하고 바로 시작하겠습니다. 먼저 아나콘다와 주피터노트북은 설치되어 있다고 가정하겠습니다. GPU가 인식되는 Tensorflow 사용을 위해서는 아래 세 가지 조건이 충족되어야 합니다. (순서대로 설치하는 것을 추천)

 

Python 3.11 이하 버전

Numpy 1.x 버전

Tensorflow 2.10 이하 버전

 

먼저 Anaconda Prompt로 접속을 합니다. 다음으로 가상환경을 구축하며 파이썬 3.9 버전으로 설정합니다. 

 

conda create -n tensorflow-env python=3.9

 

<tensorflow-env>는 가상환경 이름으로 알맞게 바꾸셔도 됩니다. 다음, 만든 가상환경을 활성화 시킵니다.

 

conda activate tensorflow-env

 

넘파이 2.0이 넘는 최신버전은 Tensorflow 2.10 버전 이상과 호환이 되지 않습니다. Numpy는 기본적으로 깔려 있기에 다운그레이드를 해줘야 합니다.

 

pip install --upgrade "numpy<2"

 

그리고 Tensorflow를 설치해 줍니다.

 

pip install "tensorflow<2.11"

 

cudatoolkit와 cudnn을 설치해 줍니다.

 

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

 

마지막으로 다음 명령어로 GPU의 인식여부를 확인합니다. 

 

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

 

만약 빈 괄호([])가 아니라 아래처럼 나온다면 성공입니다.

 

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]