SKCONAN

วิธีการติดตั้ง CUDA บน Windows 10 เพื่อใช้งาน TensorFlow-GPU

February 21, 2019

title

CUDA GPUs คืออะไร

CUDA เป็นซอร์ฟแวร์ที่ NVIDIA พัฒนาขึ้นมาเพื่อให้เราสามารถทำการคำนวณต่างๆแบบ parallel ได้ ทำให้เราลดเวลาในการประมวลผลได้โดยการย้ายการคำนวณให้มาทำงานบน GPU แทน ในที่นี้ก็คือ แทนที่เราจะรัน TensorFlow บน CPU เราก็มารันบน GPU แทน แล้วก็ยังมี cuDNN อีกตัวนึงที่ต้องใช้ร่วมกัน ซึ่งเป็น library สำหรับการทำ deep neural network

ก่อนจะติดตั้งให้เช็คก่อนนะครับว่า GPU เรารองรับการติดตั้ง CUDA หรือเปล่า CUDA-Enabled Products

ขั้นตอนการติดตั้ง

โปรแกรมที่เราจะลงมี 3 ตัวก็ คือ

  • CUDA 9.0
  • cuDNN 7.4.2
  • Python 3.6 + TensorFlow

การติดตั้ง CUDA

cuda download

ก็ให้เราเข้าไปโหลดโปรแกรมที่ CUDA Toolkit 9.0 Downloads นะครับ โดยให้คลิกเลือกตามรูปด้านบนเลยครับ ตัวติดตั้งจะประกอบไปด้วย Base Installer 1 ไฟล์ แล้วก็ไฟล์ Patch 4 ไฟล์ ให้โหลดมาให้หมดเลยนะครับ ระหว่างนี้ก็แอบไปโหลด cuDNN กันไว้ก่อนก็ได้นะครับ

cuda install

อ่าเมื่อดาวน์โหลดเสร็จแล้วก็ให้ติดตั้ง cuda_9.0.176_win10.exe ก่อนนะครับ ติดตั้งเหมือนโปรแกรมทั่วไปเลย เลือกค่า default ทั้งหมดแล้วก็กด Next ไปจนเสร็จ

cuda patch

ขั้นตอนต่อมาก็ให้ไล่ลง Patch ตั้งแต่ Patch ที่ 1 - 4 นะครับ เมื่อลงโปรแกรมน้เสร็จ ก็มาตั้งค่า PATH ของเราต่อกันดีกว่า

path 00
  1. ให้คลิกขวาที่ This PC และเลือก Properties
path 01
  1. ตรง Layout ด้านซ้ายให้เลือก Advanced system settings >> Environment Variables...

  2. ดับเบิลคลิกที่ Path —(3) หลังจากนั้นให้คลิก New แล้วพิมพ์ path ของ Folder ที่มีไฟล์ cudart64_90.dll อยู่ ของผมก็จะเป็น C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin แล้วก็เข้าไปเพิ่มที่ PATH ข้างบนด้วยนะครับ สรุปเราเพิ่ม path ไป 2 ที่นะครับ

ที่เราต้องเพิ่ม PATH ตรงนี้เข้าไป เพราะ ว่าเวลาเราใช้งาน tensorflow-gpu มันจะหาไฟล์ cudart64_90.dll จาก Environment variables

ถ้าเราไม่เพิ่ม PATH เวลาใช้งาน tensorflow-gpu ก็จะขึ้น Error ตามด้านล่างนี้

ImportError: Could not find 'cudart64_90.dll'. TensorFlow 
requires that this DLL be installed in a directory that is
named in your %PATH% environment variable. Download and 
install CUDA 9.0 from this URL: 
https://developer.nvidia.com/cuda-toolkit

การติดตั้ง cuDNN

cudnn download

ก็ให้เราเข้าไปโหลดโปรแกรมที่ cuDNN Download นะครับ โดยเลือก Download cuDNN v7.4.2 (Dec 14, 2018), for CUDA 9.0 ตามรูปด้านบนเลยครับ

เมื่อดาวน์โหลดเสร็จแล้วให้ทำการ Extract file เราจะได้ Folder cuda ที่ประกอบไปด้วย 3 Folder คือ bin, include, และ lib

cudnn copy

ให้เรา Copy 3 folder นี้ไปวางที่ C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 แค้นี้ก็เสร็จแล้วครับ ต่อไปไปดู วิธีการติดตั้ง TensorFlow-GPU ผ่าน pip (Python) กันครับ

หากใครชอบเนื้อหาที่ผมเขียน สามารถร่วมกัน Donate เพื่อเป็นค่ากาแฟได้นะครับ ^^


ใครที่สงสัยจุดไหน หรือพบว่าจุดไหนที่ผมอธิบายผิด สามารถเข้ามาพูดคุยกันได้นะครับ inbox มาที่ Facebook, Twitter เลยก็ได้ครับ หรือ mail มาที่ supakit.kr@gmail.com จะยินดีมากเลยครับ



รับพัฒนาโปรแกรมทางด้าน Image Processing, Computer Vision และ Machine Learning สามารถติดต่อ สอบถาม และปรึกษาได้ฟรีครับ

Written by Supakit Kriangkhajorn

© 2019, Built with