(윈도우) 텐서플로우(GPU)를 설치해보자 (아나콘다)

CPU를 사용해야 하는 상황이라면

pip install tensorflow

import tensorflow as tf
tf.config.list_physical_devices()

이거면 설치가 끝납니다.

 

그러나 GPU 를 사용해야 하는 상황에서 설치방법을 적으려 합니다.

(모든게 귀찮거나 그래픽카드가 없으면  그냥 colab으로 가면됩니다.)

pip로 TensorFlow 설치

pip로 TensorFlow 설치

언제나 내 글보다는 Documentaion 이 정확하다.

 

 

Official Drivers | NVIDIA

Download the latest official NVIDIA drivers

 

그래픽카드가 있는 사람중에서 설치가 안된 사람이 있을까 싶지만 

드라이버는 여기에 들어가서 다운로드 할 수 있다.

(문제는 tensorflow 환경을 구축해놓은 후에 그래픽카드 드라이버를 자동으로 업데이트 해야하는가 이게 문제..)

 

 

공식은 mini conda를 사용하지만

나는 아나콘다를 사용하고 있기 때문에 아나콘다로 바꿔서 작성하도록 하겠다.

 

우선 가상환경을 만들어서 접근하는 것을 추천한다.

 

conda create -n 이름 python=3.8

공식에서는 python3.9를 사용합니다.

3.9로 진행하셔도 될 듯 합니다.

conda activate 이름

(나는 일단 지금의 가상환경에 만들어보겠습니다.)

 

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

설치가 진행되는 모습

pip는 가상환경에 conda는 모든 환경에 설치시키는 느낌으로 보입니다.

 

 

혹시 모르니 다시 한 번 실행해보세요.

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

실행하니 

이렇게 나와서 아나콘다를 업데이트 시켰습니다.

 

오류가 나서 보니 아나콘다를 업데이트 시키라는것으로 보입니다.

conda update -n base -c defaults conda

 

 

 

 

그리고 다시 해보니 설치가 잘되어있다.

 

이제 tensorflow를 설치한다.

pip install --upgrade pip

pip install "tensorflow<2.11"

 

다음을 입력했을때 장치목록이 반환되면 성공적으로 설치된것입니다.

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

 

나는 안되니까 쥬피터로 확인을 해보겠스빈다.

import tensorflow as tf
tf.config.list_physical_devices("GPU")

import tensorflow as tf
tf.config.list_physical_devices("GPU")
tf.__version__
from tensorflow.python.client import device_lib
device_lib.list_local_devices()

 

이렇게 나오면 설치가 정상적으로 된 것입니다.

 

리눅스의 경우 더 복잡해지는데 그건 나중에 해보도록하죠,