การติดตั้ง TensorFlow GPU v. 2.4.1 Python 3.8 บน Windows 10 แบบง่าย

บทความนี้เป็นการแนะนำการติดตั้ง Tensorflow GPU 2.4.1 บน Windows 10 ว่ามีขั้นตอนอย่างไรบ้าง แบบ step by step ไปเริ่มกันเลย

สิ่งที่ต้องติดตั้งมีอะไรบ้าง

• Ananconda Python 3.8
• Microsoft Visual C++ 2015–2019 Redistributable (x64)
• NVIDIA CUDA
• cuDNN
• Tensorflow GPU

1). ติดตั้ง Anaconda บน Windows ตามรูปที่ 1. (Link download ตัวติดตั้ง)

1. Download ตัวติดตั้งตาม os ที่เราใช้งานอยู่ในที่นี้เราเลือก Windows 64-Bit Graphical Installer

2). ติดตั้ง Microsoft Visual C++ 2015–2019 Redistributable (x64) ตามรูปที่ 2. (Link Download)

2. Microsoft Visual C++ 2015–2019 Redistributable (x64)

3). ติดตั้ง CUDA และ cuDNN

check cudaversion command: nvcc — version
ก่อนอื่นให้เราไปที่ https://www.tensorflow.org/install/source_windows
ตามรูปที่ 3. เพื่อตรวจสอบเวอร์ชันของ CUDA และ cuDNN ที่ต้องติดตั้งสำหรับ Tensorflow_gpu-2.4.0

3. CUDA และ cuDNN ของ tensorflow-gpu แต่ละเวอร์ชัน

จากนั้นทำการสร้าง Environment ใหม่ โดยเปิด Anaconda Prompt แล้วพิมพ์ command ตามรูปที่ 4.

conda create -n tfgpu python==3.8

4. สร้าง environment tfgpu เพื่อรัน tensorflow gpu

จากนั้นให้ทำการ activate environment ที่เพิ่งสร้างด้วยคำสั่ง จะได้หน้าตาตามรูปที่ 5.

conda activate tfgpu

5. เข้าใช้งาน env ที่สร้างขึ้น

ทำการติดตั้ง CUDA และ cuDNN ด้วยคำสั่ง

conda install cudatoolkit=11.0 cudnn=8.0 -c=conda-forge

6. ติดตั้ง CUDA version 11.0 และ cuDNN version 8.0

4). Install Tensorflow GPU 2.4.1 ด้วยคำสั่ง

pip install --upgrade tensorflow-gpu==2.4.1

5). ทดสอบการติดตั้ง ให้พิมพ์คำสั่งต่อไปนี้ในหน้า dos prompt

(tfgpu) C:\Users\amorn>python
>>> import tensorflow as tf
>>> tf.test.is_gpu_available()

หากทุกอย่างถูกต้องจะได้ผลลัพธ์เป็น ชื่อการ์ดจอที่เราใช้ในที่นี้คือ GeForce GTX 1650 และผลรับ True ดังรูปที่ 7.

7. ผลลัพธ์การทดสอบการติดตั้ง tensorflow gpu บน windows 10

หากบทความนี้พอมีประโยชน์อยู่บ้าง ขอกำลังใจ กดปรบมือให้ด้วยนะครับ

ขอบคุณครับ

Reference:
https://www.tensorflow.org/install/source_windows
https://www.youtube.com/watch?v=toJe8ZbFhEc