Το PyTorch μπορεί να επιταχύνει τις εφαρμογές AI/ML χρησιμοποιώντας μια GPU NVIDIA μέσω της βιβλιοθήκης NVIDIA CUDA εγγενώς όπως το TensorFlow.
Σε αυτό το άρθρο, θα σας δείξουμε πώς να εγκαταστήσετε το PyTorch με υποστήριξη επιτάχυνσης NVIDIA GPU/CUDA στο Debian 12 'Bookworm'.
Θέμα Περιεχομένων:
- Εγκατάσταση των προγραμμάτων οδήγησης GPU NVIDIA στο Debian 12
- Εγκατάσταση του NVIDIA CUDA στο Debian 12
- Εγκατάσταση Python 3 PIP και Python 3 Virtual Environment (venv) στο Debian 12
- Δημιουργία εικονικού περιβάλλοντος Python 3 για PyTorch
- Αναβάθμιση Python 3 PIP στην πιο πρόσφατη έκδοση στο εικονικό περιβάλλον Python 3 PyTorch
- Εγκατάσταση του PyTorch με NVIDIA GPU/CUDA Acceleration Support στο Debian 12
- Ενεργοποίηση του εικονικού περιβάλλοντος PyTorch Python 3
- Πρόσβαση στο PyTorch και έλεγχος εάν η επιτάχυνση GPU/CUDA NVIDIA είναι διαθέσιμη
- συμπέρασμα
Εγκατάσταση των προγραμμάτων οδήγησης GPU NVIDIA στο Debian 12
Για να λειτουργήσει η επιτάχυνση GPU/CUDA NVIDIA PyTorch, πρέπει εγκαταστήστε τα προγράμματα οδήγησης NVIDIA GPU στο Debian 12 . Εάν χρειάζεστε βοήθεια για την εγκατάσταση των προγραμμάτων οδήγησης GPU NVIDIA στο σύστημά σας 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) στο Debian 12
Για να εγκαταστήσετε το PyTorch στο Debian 12, πρέπει να έχετε εγκατεστημένο το Python 3 PIP και το εικονικό περιβάλλον Python (venv).
Αρχικά, ενημερώστε τη μνήμη cache του αποθετηρίου πακέτου APT με την ακόλουθη εντολή:
$ sudo εύστοχη ενημέρωση 
Για να εγκαταστήσετε το Python 3 PIP και το εικονικό περιβάλλον Python 3 (venv), εκτελέστε την ακόλουθη εντολή:
$ sudo κατάλληλος εγκαθιστώ python3-pip python3-venv python3-devΓια να επιβεβαιώσετε την εγκατάσταση, πατήστε 'Y' και στη συνέχεια πατήστε <Εισαγωγή> .
Τα Python 3 PIP και Python 3 venv εγκαθίστανται. Χρειάζεται λίγος χρόνος για να ολοκληρωθεί.
Σε αυτό το σημείο, θα πρέπει να εγκατασταθούν Python 3 PIP και Python 3 venv.
Δημιουργία εικονικού περιβάλλοντος Python 3 για PyTorch
Η τυπική πρακτική για την εγκατάσταση των βιβλιοθηκών Python στο Debian 12 είναι η εγκατάστασή τους σε εικονικό περιβάλλον Python έτσι ώστε να μην παρεμβαίνουν στα πακέτα/βιβλιοθήκες Python του συστήματος.
Για να δημιουργήσετε ένα νέο εικονικό περιβάλλον Python 3 για το PyTorch στον κατάλογο '/opt/pytorch', εκτελέστε την ακόλουθη εντολή:
$ sudo python3 -Μ venv / επιλέγω / pytorchΑναβάθμιση Python 3 PIP στην πιο πρόσφατη έκδοση στο εικονικό περιβάλλον Python 3 PyTorch
Για να αναβαθμίσετε το Python 3 PIP στην πιο πρόσφατη έκδοση στο εικονικό περιβάλλον Python 3 “/opt/pytorch”, εκτελέστε την ακόλουθη εντολή:
$ sudo / επιλέγω / pytorch / αποθήκη / pip3 εγκαθιστώ --αναβαθμίζω κουκούτσι 
Εγκατάσταση του PyTorch με NVIDIA GPU/CUDA Acceleration Support στο Debian 12
Για να λειτουργήσει η επιτάχυνση PyTorch NVIDIA GPU/CUDA, πρέπει να εγκαταστήσετε τη σωστή έκδοση του PyTorch που υποστηρίζει την έκδοση του προγράμματος οδήγησης NVIDIA CUDA που εγκαταστήσατε στο σύστημά σας Debian 12. Τη στιγμή που γράφεται αυτό το άρθρο, η PyTorch υποστηρίζει τις εκδόσεις 11.8 και 12.1 του προγράμματος οδήγησης NVIDIA CUDA. Για ενημερωμένες πληροφορίες σχετικά με τις εκδόσεις του προγράμματος οδήγησης NVIDIA CUDA που υποστηρίζει η PyTorch, ελέγξτε τον επίσημο ιστότοπο του PyTorch .
Για να ελέγξετε την έκδοση του προγράμματος οδήγησης NVIDIA CUDA που εγκαταστήσατε στο σύστημά σας Debian 12, εκτελέστε την ακόλουθη εντολή. Όπως μπορείτε να δείτε, έχουμε εγκατεστημένη την έκδοση 11.8 NVIDIA CUDA στο σύστημα Debian 12.
$ nvcc --εκδοχή 
Για να εγκαταστήσετε το PyTorch με υποστήριξη NVIDIA CUDA 11.8 στο εικονικό περιβάλλον PyTorch Python 3, εκτελέστε την ακόλουθη εντολή:
$ sudo / επιλέγω / pytorch / αποθήκη / pip3 εγκαθιστώ πυρσός torchvision torchaudio --index-url https: // download.pytorch.org / whl / με 118Για να εγκαταστήσετε το PyTorch με υποστήριξη NVIDIA CUDA 12.1 στο εικονικό περιβάλλον PyTorch Python 3, εκτελέστε την ακόλουθη εντολή:
$ sudo / επιλέγω / pytorch / αποθήκη / pip3 εγκαθιστώ πυρσός torchvision torchaudioΤο PyTorch εγκαθίσταται στο εικονικό περιβάλλον PyTorch Python 3. Χρειάζεται λίγος χρόνος για να ολοκληρωθεί.
Σε αυτό το σημείο, το PyTorch θα πρέπει να εγκατασταθεί στο εικονικό περιβάλλον PyTorch Python 3
Ενεργοποίηση εικονικού περιβάλλοντος PyTorch Python 3
Για να ενεργοποιήσετε το εικονικό περιβάλλον PyTorch Python '/opt/pytorch', εκτελέστε την ακόλουθη εντολή:
$ . / επιλέγω / pytorch / αποθήκη / δραστηριοποιώΤο εικονικό περιβάλλον PyTorch Python 3 θα πρέπει να ενεργοποιηθεί.
Πρόσβαση στο PyTorch και έλεγχος εάν η επιτάχυνση GPU/CUDA NVIDIA είναι διαθέσιμη
Για να ανοίξετε το διαδραστικό κέλυφος της Python 3, εκτελέστε την ακόλουθη εντολή:
$ python3Θα πρέπει να ανοίξει το διαδραστικό κέλυφος Python 3.
Πρώτα, εισαγάγετε το PyTorch με την ακόλουθη γραμμή κώδικα:
$ φακός εισαγωγής 
Για να ελέγξετε την έκδοση του PyTorch που εγκαταστήσατε, εκτελέστε την ακόλουθη γραμμή κώδικα. Όπως μπορείτε να δείτε, εκτελούμε το PyTorch 2.1.0 με υποστήριξη επιτάχυνσης NVIDIA CUDA 11.8 (cu118).
$ φακός.__έκδοση__ 
Για να ελέγξετε εάν το PyTorch μπορεί να χρησιμοποιήσει τη GPU NVIDIA για επιτάχυνση NVIDIA CUDA, μπορείτε επίσης να εκτελέσετε την ακόλουθη γραμμή κώδικα. Εάν η υποστήριξη NVIDIA CUDA είναι διαθέσιμη, θα εκτυπωθεί το 'True'.
$ torch.cuda.is_available ( )Εάν έχετε εγκαταστήσει πολλές GPU στον υπολογιστή σας, μπορείτε να ελέγξετε τον αριθμό των GPU που μπορεί να χρησιμοποιήσει η PyTorch με την ακόλουθη γραμμή κώδικα. Όπως μπορείτε να δείτε, έχουμε την NVIDIA GPU (RTX 4070) εγκατεστημένη στο σύστημά μας Debian 12.
$ torch.cuda.device_count ( )Για έξοδο από το διαδραστικό κέλυφος της Python, εκτελέστε την ακόλουθη γραμμή κώδικα:
$ εγκαταλείπω ( )συμπέρασμα
Σε αυτό το άρθρο, σας δείξαμε πώς να εγκαταστήσετε το Python 3 PIP και το εικονικό περιβάλλον Python 3 (venv) στο Debian 12. Σας δείξαμε επίσης πώς να δημιουργήσετε ένα εικονικό περιβάλλον Python 3 για το PyTorch στο Debian 12 και πώς να εγκαταστήσετε το PyTorch με το NVIDIA CUDA Υποστήριξη επιτάχυνσης 11.8 και 12.1 και στο Debian 12. Τέλος, σας δείξαμε πώς να ενεργοποιήσετε το εικονικό περιβάλλον PyTorch Python και να αποκτήσετε πρόσβαση στο PyTorch στο Debian 12.