PyTorch יכולה להאיץ את יישומי AI/ML באמצעות NVIDIA GPU דרך ספריית NVIDIA CUDA באופן מקורי בדיוק כמו TensorFlow.
במאמר זה, נראה לך כיצד להתקין PyTorch עם תמיכה בהאצת NVIDIA GPU/CUDA על Debian 12 'תולעת ספרים'.
נושא התוכן:
- התקנת מנהלי ההתקן של NVIDIA GPU ב-Debian 12
- התקנת NVIDIA CUDA ב-Debian 12
- התקנת Python 3 PIP ו- Python 3 Virtual Environment (venv) בדביאן 12
- יצירת סביבה וירטואלית של Python 3 עבור PyTorch
- שדרוג Python 3 PIP לגרסה האחרונה בסביבה הוירטואלית Python 3 PyTorch
- התקנת PyTorch עם תמיכת NVIDIA GPU/CUDA Acceleration ב-Debian 12
- הפעלת הסביבה הווירטואלית PyTorch Python 3
- גישה ל- PyTorch ובדיקה אם האצת NVIDIA GPU/CUDA זמינה
- סיכום
התקנת מנהלי ההתקן של NVIDIA GPU ב-Debian 12
כדי שהאצת PyTorch NVIDIA GPU/CUDA תעבוד, אתה חייב התקן את מנהלי ההתקן של NVIDIA GPU ב-Debian 12 . אם אתה זקוק לסיוע כלשהו בהתקנת מנהלי ההתקן של NVIDIA GPU במערכת Debian 12 שלך, לקרוא את המאמר הזה .
התקנת NVIDIA CUDA ב-Debian 12
כדי שהאצת PyTorch NVIDIA GPU/CUDA תעבוד על Debian 12, אתה חייב התקן את NVIDIA CUDA ב-Debian 12 . אם אתה זקוק לסיוע כלשהו בהתקנת NVIDIA CUDA במערכת Debian 12 שלך, לקרוא את המאמר הזה .
התקנת Python 3 PIP ו- Python 3 Virtual Environment (venv) בדביאן 12
כדי להתקין את PyTorch ב-Debian 12, עליך להתקין את Python 3 PIP ואת הסביבה הוירטואלית של Python (venv).
ראשית, עדכן את המטמון של מאגר חבילת APT בפקודה הבאה:
$ סודו עדכון מתאים 
כדי להתקין Python 3 PIP ו- Python 3 סביבה וירטואלית (venv), הפעל את הפקודה הבאה:
$ סודו מַתְאִים להתקין python3-pip python3-venv python3-dev כדי לאשר את ההתקנה, לחץ על 'Y' ולאחר מכן לחץ על
Python 3 PIP ו-Python 3 venv נמצאים בהתקנה. זה לוקח זמן להשלים.
בשלב זה, יש להתקין Python 3 PIP ו-Python 3 venv.
יצירת סביבה וירטואלית של Python 3 עבור PyTorch
הנוהג המקובל להתקנת ספריות Python בדביאן 12 הוא התקנתן בסביבה וירטואלית של Python כך שהן לא יפריעו לחבילות/ספריות Python של המערכת.
כדי ליצור סביבה וירטואלית חדשה של Python 3 עבור PyTorch בספריית '/opt/pytorch', הפעל את הפקודה הבאה:
$ סודו פיתון3 -M venv / העדיף / pytorchשדרוג Python 3 PIP לגרסה האחרונה בסביבה הוירטואלית Python 3 PyTorch
כדי לשדרג את Python 3 PIP לגרסה העדכנית ביותר בסביבה הווירטואלית '/opt/pytorch' של Python 3, הפעל את הפקודה הבאה:
$ סודו / העדיף / pytorch / פַּח / pip3 להתקין --שדרוג צִפצוּף 
התקנת PyTorch עם תמיכת NVIDIA GPU/CUDA Acceleration ב-Debian 12
כדי שההאצת PyTorch NVIDIA GPU/CUDA תעבוד, עליך להתקין את הגרסה הנכונה של PyTorch התומכת בגרסת מנהל ההתקן של NVIDIA CUDA שהתקנת במערכת Debian 12 שלך. בזמן כתיבת שורות אלה, PyTorch תומך במנהלי ההתקן של NVIDIA CUDA בגרסאות 11.8 ו-12.1. למידע מעודכן על גרסאות מנהל ההתקן של NVIDIA CUDA שבהן PyTorch תומך, בדוק את האתר הרשמי של PyTorch .
כדי לבדוק את גרסת מנהל ההתקן של NVIDIA CUDA שהתקנת במערכת Debian 12 שלך, הפעל את הפקודה הבאה. כפי שאתה יכול לראות, מותקנת אצלנו NVIDIA CUDA גרסה 11.8 במערכת Debian 12 שלנו.
$ nvcc --גִרְסָה 
כדי להתקין PyTorch עם תמיכה ב-NVIDIA CUDA 11.8 בסביבה הווירטואלית PyTorch Python 3, הפעל את הפקודה הבאה:
$ סודו / העדיף / pytorch / פַּח / pip3 להתקין torch torchvision torchaudio --index-url https: // download.pytorch.org / whl / עם 118כדי להתקין PyTorch עם תמיכה ב-NVIDIA CUDA 12.1 בסביבה הווירטואלית PyTorch Python 3, הפעל את הפקודה הבאה:
$ סודו / העדיף / pytorch / פַּח / pip3 להתקין torch torchvision torchaudioPyTorch מותקן בסביבה הווירטואלית PyTorch Python 3. זה לוקח זמן להשלים.
בשלב זה, PyTorch צריך להיות מותקן בסביבה הוירטואלית PyTorch Python 3
הפעלת PyTorch Python 3 סביבה וירטואלית
כדי להפעיל את הסביבה הוירטואלית PyTorch Python '/opt/pytorch', הפעל את הפקודה הבאה:
$ . / העדיף / pytorch / פַּח / לְהַפְעִיליש להפעיל את הסביבה הוירטואלית PyTorch Python 3.
גישה ל- PyTorch ובדיקה אם האצת NVIDIA GPU/CUDA זמינה
כדי לפתוח את המעטפת האינטראקטיבית של Python 3, הפעל את הפקודה הבאה:
$ פיתון3יש לפתוח מעטפת אינטראקטיבית של Python 3.
ראשית, ייבא את PyTorch עם שורת הקוד הבאה:
$ לייבא לפיד 
כדי לבדוק את הגרסה של PyTorch שהתקנת, הפעל את שורת הקוד הבאה. כפי שאתה יכול לראות, אנו מריצים PyTorch 2.1.0 עם תמיכה בהאצה של NVIDIA CUDA 11.8 (cu118).
$ לפיד.__גרסה__ 
כדי לבדוק אם PyTorch מסוגל להשתמש ב-NVIDIA GPU שלך עבור האצת NVIDIA CUDA, אתה יכול להריץ גם את שורת הקוד הבאה. אם תמיכת NVIDIA CUDA זמינה, 'True' יודפס.
$ torch.cuda.is_available ( )אם יש לך מספר GPUs מותקנים במחשב שלך, אתה יכול לבדוק את מספר ה-GPUs שבהם PyTorch יכול להשתמש עם שורת הקוד הבאה. כפי שאתה יכול לראות, יש לנו את NVIDIA GPU (RTX 4070) מותקן במערכת Debian 12 שלנו.
$ torch.cuda.device_count ( )כדי לצאת מהמעטפת האינטראקטיבית של Python, הפעל את שורת הקוד הבאה:
$ לְהַפְסִיק ( )סיכום
במאמר זה, הראינו לך כיצד להתקין את Python 3 PIP ו- Python 3 (venv) ב-Debian 12. הראינו לך גם כיצד ליצור סביבה וירטואלית Python 3 עבור PyTorch בדביאן 12 וכיצד להתקין PyTorch עם NVIDIA CUDA תמיכה בהאצת 11.8 ו-12.1 גם בדביאן 12. לבסוף, הראינו לך כיצד להפעיל את הסביבה הווירטואלית PyTorch Python ולגשת ל- PyTorch בדביאן 12.