ในคู่มือนี้ ฉันจะทำการติดตั้ง PyTorch บน Windows และ Linux โดยใช้ Anaconda เพื่อจัดการการติดตั้งแพ็คเกจ
การติดตั้งจะดำเนินการโดยใช้เครื่องมือบรรทัดคำสั่ง conda ที่สร้างไว้ใน Anaconda
PyTorch เป็นไลบรารีการเรียนรู้ของเครื่องที่เขียนด้วย Python และอิงตามเฟรมเวิร์กของ Torch ได้รับการพัฒนาโดย Facebook และเปรียบได้กับ Tensorflow โดย Google มีประโยชน์ในด้านการมองเห็นด้วยคอมพิวเตอร์และการประมวลผลภาษาธรรมชาติ และบริษัทต่างๆ เช่น Tesla ได้นำไปใช้เพื่อพัฒนาซอฟต์แวร์ออโตไพลอต
PyTorch เป็นโอเพ่นซอร์สฟรีและได้รับอนุญาตภายใต้ BSD ที่แก้ไขแล้วและอยู่ภายใต้ Linux Foundation
ข้อกำหนดเบื้องต้น
เพื่อทำตามบทช่วยสอนนี้ คุณต้องติดตั้งอนาคอนด้าในเครื่องที่คุณกำลังใช้งาน
หากคุณยังไม่ได้ติดตั้งไว้ คำแนะนำเกี่ยวกับวิธีการติดตั้งอนาคอนด้านี้จะแนะนำคุณตลอดกระบวนการทั้งหมด หลังจากทำตามคำแนะนำนั้นแล้ว คุณสามารถดำเนินการติดตั้ง PyTorch ได้
การติดตั้ง PyTorch บน Linux
ตามแนวทางปฏิบัติที่ดี ให้เริ่มต้นด้วยการอัปเดตแพ็คเกจซอฟต์แวร์ใน Linux distribution ของคุณ ในกรณีของฉัน ฉันใช้ Ubuntu และมักจะจัดการแพ็คเกจของฉัน ดังนั้นฉันจะใช้คำสั่งต่อไปนี้เพื่ออัปเดต:
$ sudo apt update && apt upgrade
เมื่อคุณอัปเดตแพ็คเกจเสร็จแล้ว ให้ไปที่หน้าการติดตั้งเว็บไซต์ PyTorch อย่างเป็นทางการ เลื่อนลงมาที่หน้าจนกว่าคุณจะพบวิซาร์ดการติดตั้งที่มีลักษณะดังนี้:
เมื่อใช้วิซาร์ดนี้ คุณจะสามารถคลิกที่ตัวเลือกต่างๆ เพื่อให้ข้อมูลระบบและค่ากำหนดของคุณ และในทางกลับกัน คุณจะได้รับคำสั่งที่คุณสามารถใช้ในเทอร์มินัลของคุณเพื่อติดตั้ง PyTorch
หลังจากให้ข้อมูลระบบของฉันแล้ว นี่คือสิ่งที่ดูเหมือน:
ฉันเลือกใช้เวอร์ชัน Linux ที่เสถียร และฉันจะใช้ Conda เพื่อจัดการแพ็คเกจของฉัน ฉันยังเลือกใช้ PyTorch กับ Python มากกว่า C++/Java และฉันจะใช้งาน PyTorch บน CPU แทน GPU
ที่ด้านล่างของตารางคือคำสั่งที่ฉันสามารถใช้ติดตั้ง PyTorch ได้ แต่ก่อนที่จะเรียกใช้คำสั่งนี้ ฉันต้องการสร้างสภาพแวดล้อมเสมือนของ Anaconda ที่เรียกว่า pytorch
สภาพแวดล้อมเสมือนช่วยให้คุณสร้างโปรเจ็กต์และแยกการขึ้นต่อกันออกจากการขึ้นต่อกันของโปรเจ็กต์อื่นๆ ได้ จึงเป็นการป้องกันความขัดแย้งในการขึ้นต่อกัน ข้อดีอย่างหนึ่งของ Anaconda คือช่วยให้คุณสร้างและจัดการสภาพแวดล้อมเสมือนจริงได้อย่างง่ายดาย
ในการสร้างสภาพแวดล้อมเสมือนที่เวอร์ชัน Python คือ 3.7 ฉันจะป้อนคำสั่งต่อไปนี้:
conda create -n pytorch python=3.7
หลังจากสร้างสภาพแวดล้อมแล้ว ฉันจะเปิดใช้งานโดยใช้คำสั่งต่อไปนี้:
conda activate pytorch
เมื่อสภาพแวดล้อมทำงาน ฉันจะเรียกใช้คำสั่งที่สร้างก่อนหน้านี้บนเว็บไซต์ PyTorch เพื่อติดตั้ง PyTorch
conda install pytorch torchvision torchaudio cpuonly -c pytorch
ทำตามคำแนะนำเพื่อติดตั้ง PyTorch เมื่อเสร็จแล้ว ฉันจะรีสตาร์ทเซสชันเทอร์มินัลเพื่อให้มีผล
ตอนนี้เพื่อตรวจสอบว่า PyTorch ได้รับการติดตั้งอย่างถูกต้อง เราจะพยายามนำเข้ามันใน Python แบบโต้ตอบเชลล์ ตรวจสอบให้แน่ใจว่าคุณอยู่ในสภาพแวดล้อมเสมือน Pytorch โดยใช้คำสั่ง:
conda activate pytorch
เมื่อคุณอยู่ในสภาพแวดล้อมเสมือนของ Pytorch ให้เปิดเชลล์แบบโต้ตอบของ python โดยพิมพ์คำสั่ง:
python
เมื่อเซสชันของเชลล์เริ่มต้น ให้เขียนโค้ดบรรทัดต่อไปนี้แล้วกด ENTER
import torch
หาก Python ทำงานโดยไม่มีข้อผิดพลาด แสดงว่าการติดตั้งสำเร็จ แต่ถ้าคุณได้รับข้อผิดพลาดไม่พบโมดูล แสดงว่ามีบางอย่างผิดพลาดระหว่างการติดตั้ง คุณอาจลองติดตั้งใหม่อีกครั้ง
การติดตั้ง PyTorch บน Windows
ในการเริ่มต้น บนเครื่อง Windows ของคุณ ให้ค้นหาโปรแกรม Anaconda Prompt แล้วเปิดขึ้นมา นี่คือที่ที่เราจะเรียกใช้คำสั่ง
เมื่อเปิดโปรแกรมแล้ว เราจะสร้างสภาพแวดล้อมเสมือนสำหรับการติดตั้ง PyTorch โดยใช้คำสั่ง
conda create -n pytorch python=3.7
หลังจากสร้างสภาพแวดล้อมเสมือนแล้ว เราสามารถเปิดใช้งานได้โดยใช้คำสั่งต่อไปนี้:
conda activate pytorch
เมื่อสภาพแวดล้อมเสมือนทำงาน เราสามารถดำเนินการติดตั้ง PyTorch ได้ เราเริ่มต้นด้วยการไปที่หน้าการติดตั้งเว็บไซต์ PyTorch หลังจากนั้น เราสามารถเลื่อนลงไปที่ส่วนของหน้าที่เป็นที่ตั้งของวิซาร์ดการติดตั้งนี้:
ที่นี่ เราเลือกข้อมูลระบบของเรา และวิซาร์ดจะให้คำสั่งติดตั้ง PyTorch ฉันจะเลือกรุ่นที่เสถียรสำหรับ Windows จัดการโดย Conda ใช้ผ่านภาษาโปรแกรม python และทำงานบน CPU เป็นผลให้ตารางของฉันจะมีลักษณะเช่นนี้
ถัดไป ให้คัดลอกคำสั่ง วางลงในพรอมต์ของ Anaconda แล้วกด ENTER
หลังจากการติดตั้งเสร็จสิ้น เราสามารถตรวจสอบว่าดำเนินการสำเร็จหรือไม่โดยเปิด Python Interactive Shell และพยายามนำเข้า PyTorch
ดังนั้นภายในพรอมต์ของ Anaconda ให้เริ่มเซสชัน Python แบบโต้ตอบ
python
หลังจากเริ่มเซสชัน ให้นำเข้า PyTorch โดยใช้รหัสบรรทัดต่อไปนี้:
import torch
หากการดำเนินการนี้เสร็จสิ้นโดยไม่มีข้อผิดพลาด แสดงว่าการติดตั้งสำเร็จ
คำพูดสุดท้าย
ในคู่มือนี้ เราได้ติดตั้ง PyTorch ทั้งบน Windows และ Linux โดยใช้ conda สามารถติดตั้งผ่าน PIP ได้เหมือนกับแพ็คเกจ PIP ปกติ ในทั้งสองกรณี ฉันเลือกใช้การติดตั้ง CPU อย่างไรก็ตาม คุณยังสามารถใช้ CUDA ซึ่งเป็นชุดเครื่องมือระบบที่พัฒนาโดย Nvidia ที่เร่งความเร็วการฝึกอบรมด้วยการขนานการทำงานข้าม GPU
