PyTorch dapat mempercepat aplikasi AI/ML menggunakan GPU NVIDIA melalui perpustakaan NVIDIA CUDA seperti TensorFlow.
Pada artikel ini, kami akan menunjukkan cara menginstal PyTorch dengan dukungan akselerasi GPU NVIDIA/CUDA di Debian 12 “Bookworm”.
Topik Isi:
- Menginstal Driver GPU NVIDIA di Debian 12
- Menginstal NVIDIA CUDA di Debian 12
- Menginstal Python 3 PIP dan Python 3 Virtual Environment (venv) di Debian 12
- Membuat Lingkungan Virtual Python 3 untuk PyTorch
- Meningkatkan PIP Python 3 ke Versi Terbaru di Lingkungan Virtual Python 3 PyTorch
- Menginstal PyTorch dengan Dukungan Akselerasi GPU NVIDIA/CUDA di Debian 12
- Mengaktifkan Lingkungan Virtual PyTorch Python 3
- Mengakses PyTorch dan Memeriksa Apakah Akselerasi GPU NVIDIA/CUDA Tersedia
- Kesimpulan
Menginstal Driver GPU NVIDIA di Debian 12
Agar akselerasi GPU/CUDA PyTorch NVIDIA berfungsi, Anda harus instal driver GPU NVIDIA di Debian 12 . Jika Anda memerlukan bantuan dalam menginstal driver GPU NVIDIA pada sistem Debian 12 Anda, membaca artikel ini .
Menginstal NVIDIA CUDA di Debian 12
Agar akselerasi GPU/CUDA PyTorch NVIDIA dapat berfungsi di Debian 12, Anda harus melakukannya instal NVIDIA CUDA di Debian 12 . Jika Anda memerlukan bantuan dalam menginstal NVIDIA CUDA pada sistem Debian 12 Anda, membaca artikel ini .
Menginstal Python 3 PIP dan Python 3 Virtual Environment (venv) di Debian 12
Untuk menginstal PyTorch di Debian 12, Anda harus menginstal PIP Python 3 dan lingkungan virtual Python (venv).
Pertama, perbarui cache repositori paket APT dengan perintah berikut:
$ sudo pembaruan yang tepat 
Untuk menginstal lingkungan virtual Python 3 PIP dan Python 3 (venv), jalankan perintah berikut:
$ sudo tepat Install python3-pip python3-venv python3-dev Untuk mengonfirmasi instalasi, tekan 'Y' lalu tekan
Python 3 PIP dan Python 3 venv sedang diinstal. Butuh beberapa saat untuk menyelesaikannya.
Pada titik ini, Python 3 PIP dan Python 3 venv harus diinstal.
Membuat Lingkungan Virtual Python 3 untuk PyTorch
Praktik standar untuk menginstal pustaka Python di Debian 12 adalah menginstalnya di lingkungan virtual Python sehingga tidak mengganggu paket/pustaka Python sistem.
Untuk membuat lingkungan virtual Python 3 baru untuk PyTorch di direktori “/opt/pytorch”, jalankan perintah berikut:
$ sudo python3 -M venv / memilih / pytorchMeningkatkan PIP Python 3 ke Versi Terbaru di Lingkungan Virtual Python 3 PyTorch
Untuk memutakhirkan PIP Python 3 ke versi terbaru di lingkungan virtual “/opt/pytorch” Python 3, jalankan perintah berikut:
$ sudo / memilih / pytorch / tempat sampah / pip3 Install --meningkatkan pip 
Menginstal PyTorch dengan Dukungan Akselerasi GPU NVIDIA/CUDA di Debian 12
Agar akselerasi GPU/CUDA PyTorch NVIDIA berfungsi, Anda harus menginstal versi PyTorch yang benar yang mendukung versi driver NVIDIA CUDA yang Anda instal pada sistem Debian 12 Anda. Pada saat penulisan ini, PyTorch mendukung driver NVIDIA CUDA versi 11.8 dan 12.1. Untuk informasi terbaru tentang versi driver NVIDIA CUDA yang didukung PyTorch, periksa situs web resmi PyTorch .
Untuk memeriksa versi driver NVIDIA CUDA yang Anda instal pada sistem Debian 12 Anda, jalankan perintah berikut. Seperti yang Anda lihat, kami telah menginstal NVIDIA CUDA versi 11.8 di sistem Debian 12 kami.
$ nvcc --Versi: kapan 
Untuk menginstal PyTorch dengan dukungan NVIDIA CUDA 11.8 di lingkungan virtual PyTorch Python 3, jalankan perintah berikut:
$ sudo / memilih / pytorch / tempat sampah / pip3 Install obor obor vision torchaudio --indeks-url https: // unduh.pytorch.org / whl / dengan 118Untuk menginstal PyTorch dengan dukungan NVIDIA CUDA 12.1 di lingkungan virtual PyTorch Python 3, jalankan perintah berikut:
$ sudo / memilih / pytorch / tempat sampah / pip3 Install obor obor vision torchaudioPyTorch sedang diinstal di lingkungan virtual PyTorch Python 3. Butuh beberapa saat untuk menyelesaikannya.
Pada titik ini, PyTorch harus diinstal pada lingkungan virtual PyTorch Python 3
Mengaktifkan Lingkungan Virtual PyTorch Python 3
Untuk mengaktifkan lingkungan virtual PyTorch Python “/opt/pytorch”, jalankan perintah berikut:
$ . / memilih / pytorch / tempat sampah / mengaktifkanLingkungan virtual PyTorch Python 3 harus diaktifkan.
Mengakses PyTorch dan Memeriksa Apakah Akselerasi GPU NVIDIA/CUDA Tersedia
Untuk membuka shell interaktif Python 3, jalankan perintah berikut:
$ python3Shell interaktif Python 3 harus dibuka.
Pertama, impor PyTorch dengan baris kode berikut:
$ impor obor 
Untuk memeriksa versi PyTorch yang Anda instal, jalankan baris kode berikut. Seperti yang Anda lihat, kami menjalankan PyTorch 2.1.0 dengan dukungan akselerasi NVIDIA CUDA 11.8 (cu118).
$ obor.__version__ 
Untuk memeriksa apakah PyTorch mampu menggunakan GPU NVIDIA Anda untuk akselerasi NVIDIA CUDA, Anda juga dapat menjalankan baris kode berikut. Jika dukungan NVIDIA CUDA tersedia, “True” akan dicetak.
$ torch.cuda.is_tersedia ( )Jika Anda memiliki beberapa GPU yang terpasang di komputer Anda, Anda dapat memeriksa jumlah GPU yang dapat digunakan PyTorch dengan baris kode berikut. Seperti yang Anda lihat, kami telah menginstal GPU NVIDIA (RTX 4070) di sistem Debian 12 kami.
$ torch.cuda.device_count ( )Untuk keluar dari shell interaktif Python, jalankan baris kode berikut:
$ berhenti ( )Kesimpulan
Pada artikel ini, kami menunjukkan kepada Anda cara menginstal lingkungan virtual (venv) Python 3 PIP dan Python 3 di Debian 12. Kami juga menunjukkan kepada Anda cara membuat lingkungan virtual Python 3 untuk PyTorch di Debian 12 dan cara menginstal PyTorch dengan NVIDIA CUDA Dukungan akselerasi 11.8 dan 12.1 pada Debian 12 juga. Terakhir, kami menunjukkan kepada Anda cara mengaktifkan lingkungan virtual PyTorch Python dan mengakses PyTorch di Debian 12.