How to install TensorFlow-GPU for Python(Easy-guide)

Narumit Kitiweth
2 min readJun 1, 2020

--

สวัสดีครับ story นี่จะเป็นการบันทึกไว้ สำหรับการติดตั้ง Tensorflow-gpu อย่างง่าย

โดยวันนี้เราจะมาเริ่มลง Tensorflow-gpu เวอร์ชั่น 2.x.x กัน

Ps.วันที่เขียนบทความ 1/6/2020 จะใช้เวอร์ชั่นตามนี้

Requirement:

  1. CUDA Toolkit 10.1 update2 (Aug 2019) หากต้องการเลือกเวอร์ชั่นอื่นสามารถเข้าได้ตามลิ้งก์นี้-> https://developer.nvidia.com/cuda-toolkit-archive
  2. Download cuDNN v7.6.5 (November 5th, 2019), for CUDA 10.1หากต้องการเลือกเวอร์ชั่นอื่นสามารถเข้าได้ตามลิ้งก์นี้(ติ๊ก I Agree Terms ก่อน)
    ->https://developer.nvidia.com/rdp/cudnn-download (ถ้ายังไม่ได้Login เมื่อLoginเสร็จ ให้กลับมากดลิ้งค์นี้อีกที)

วิธีการลง Tensorflow-gpu

pip install tensorflow-gpu โดยใช้ command line (หากไม่สามารถลงได้ ให้ลองupdate ทุกlibraryก่อน -> conda update -- all)

  1. Install CUDA Toolkit 10.1 ที่เราดาวน์โหลดมา
  2. ให้เข้าไปที่Folder ที่เราติดตั้งCUDA เช่น C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
  3. แตกไฟล์ cuDNNจากนั้นเอาไฟล์ข้างในไปวางFolderที่เราติดตั้งCUDAตามข้อ3
ลากไฟล์จากcuDNNลงทับไปได้เลย

จะเห็นได้ว่าเรารันผ่าน และเมื่อเช็คGPU ด้วย Task manager ก็จะเห็นได้ว่ามีการใช้RAM ของGPU ตามรูปเลยครับ วิธีเช็คก็ Task manager->Performance

หากว่าได้ลงไปแล้วหลายตัว สามารถเข้าไปเปลี่ยนenv ตรง sytemให้เป็นเวอร์ชั่น10.1ก่อนได้

Ps2.ถ้าต้องการลงTensorflow-gpu เวอร์ชั่น1.x.x ก็ให้ใช้ CUDAเวอร์ชั่น10.0 และให้ใช้cuDNNสำหรับ10.0 แทน

--

--

Narumit Kitiweth
Narumit Kitiweth

Written by Narumit Kitiweth

Data engineer | Data scientist | Machine learning researcher

No responses yet