안녕하세요,
재재입니다.
Tensorflow 2.4.0 설치하기 포스팅으로,
cuda 설치, cudnn 설치를 완료하셨다면 진행해주세요.
(1) Tensorflow 2.4.0 및 관련 library 설치
# install tensorflow 2.4 and related packages.
pip install tensorflow-gpu==2.4 tensorflow-probability==0.12.0 tf-nightly-gpu==2.9.0.dev20211223
pip install tensorflow_addons tensorflow_model_optimization tensorflow_datasets
pip install Pillow opencv-python
(2) Tensorflow 가 정상적으로 설치되었는지 확인
python3
일단, python 을 먼저 실행해주세요.
import tensorflow as tf
tf.__version__
(3) Tensorflow 에 맞는 cuda/cudnn 버전 확인하기
https://www.tensorflow.org/install/source?hl=ko#tested_build_configurations
tensorflow 2.4.0 설치하기