2.1. Cài TensorFlow GPU (nếu chưa cài):
pip uninstall tensorflow
pip install tensorflow==2.13
2.2. Cài thêm thư viện hỗ trợ GPU:
pip install nvidia-pyindex
pip install nvidia-cuda-runtime-cu11
2.3. Cài đặt CUDA + cuDNN
2.3.1. Kiểm tra GPU và Driver
Mở CMD: nvidia-smi
Nếu thấy bảng GPU → OK
2.3.2. Cài CUDA Toolkit 11.8
Chọn:
- OS: Windows
- Installer Type: Local / Network Installer
- Architecture: x86_64
Cài đặt đầy đủ CUDA 11.8 (không cần thêm vào PATH nếu dùng pip TensorFlow >= 2.10)
2.3.3. Cài cuDNN 8.6
Chọn cuDNN v8.6.0 for CUDA 11.x
Tải bản cuDNN Library for Windows (zip)
Giải nén và copy vào thư mục CUDA:
bin → C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
include → C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include
lib → C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib\x64
3. Kiểm tra TensorFlow đã nhận GPU chưa
Mở python và chạy thử ra kết quả là 0 thì OK
import tensorflow as tf
print("Num GPUs Available:", len(tf.config.list_physical_devices('GPU')))