windows에서 tensorflow 설치(GPU)

윈도우 native에서 마지막으로 설치 가능한 tensorflow-gpu==2.10.1 버전을 설치하는 가이드입니다.
tensorflow는 2.11버전 이후로 윈도우에서 gpu를 사용하기 위해서는 WSL2를 사용하여야 하며, tensorflow-gpu 패키지와 tensorflow 패키지가 통합됩니다.

필요 설치사항

anaconda 새로운 환경 생성

Anaconda Prompt를 실행하고 아래 명령을 넣어 tensorflow를 실행하기 위한 환경을 생성한다.

conda create -n tf pip python=3.10 ipython ipykernel
conda activate tf

tensorflow 설치

이어 아래 명령을 통해 tensorflow를 설치하고 정상적으로 설치되었는지 확인한다.

conda install -c conda-forge cudatoolkit=11.8 cudnn=8.9.1
python -m pip install "tensorflow-gpu<2.11"

Verify install

python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

conda 가상환경

가상환경 리스트 확인

conda info --envs

가상환경 삭제

conda remove --name {virtual env name} --all