PyTorch boleh mempercepatkan aplikasi AI/ML menggunakan GPU NVIDIA melalui perpustakaan NVIDIA CUDA secara asli seperti TensorFlow.
Dalam artikel ini, kami akan menunjukkan kepada anda cara memasang PyTorch dengan sokongan pecutan GPU/CUDA NVIDIA pada Debian 12 'Bookworm'.
Topik Kandungan:
- Memasang Pemacu GPU NVIDIA pada Debian 12
- Memasang NVIDIA CUDA pada Debian 12
- Memasang Persekitaran Maya Python 3 PIP dan Python 3 (venv) pada Debian 12
- Mencipta Persekitaran Maya Python 3 untuk PyTorch
- Menaik taraf Python 3 PIP kepada Versi Terkini pada Persekitaran Maya Python 3 PyTorch
- Memasang PyTorch dengan Sokongan Pecutan GPU/CUDA NVIDIA pada Debian 12
- Mengaktifkan Persekitaran Maya PyTorch Python 3
- Mengakses PyTorch dan Menyemak Jika NVIDIA GPU/CUDA Acceleration Tersedia
- Kesimpulan
Memasang Pemacu GPU NVIDIA pada Debian 12
Untuk pecutan GPU/CUDA PyTorch NVIDIA berfungsi, anda mesti pasang pemacu GPU NVIDIA pada Debian 12 . Jika anda memerlukan sebarang bantuan dalam memasang pemacu GPU NVIDIA pada sistem Debian 12 anda, baca artikel ini .
Memasang NVIDIA CUDA pada Debian 12
Untuk pecutan PyTorch NVIDIA GPU/CUDA berfungsi pada Debian 12, anda mesti pasang NVIDIA CUDA pada Debian 12 . Jika anda memerlukan sebarang bantuan dalam memasang NVIDIA CUDA pada sistem Debian 12 anda, baca artikel ini .
Memasang Persekitaran Maya Python 3 PIP dan Python 3 (venv) pada Debian 12
Untuk memasang PyTorch pada Debian 12, anda perlu memasang Python 3 PIP dan persekitaran maya Python (venv).
Mula-mula, kemas kini cache repositori pakej APT dengan arahan berikut:
$ sudo kemas kini yang sesuai 
Untuk memasang persekitaran maya Python 3 PIP dan Python 3 (venv), jalankan arahan berikut:
$ sudo apt pasang python3-pip python3-venv python3-dev Untuk mengesahkan pemasangan, tekan 'Y' dan kemudian tekan
Python 3 PIP dan Python 3 venv sedang dipasang. Ia mengambil sedikit masa untuk disiapkan.
Pada ketika ini, Python 3 PIP dan Python 3 venv harus dipasang.
Mencipta Persekitaran Maya Python 3 untuk PyTorch
Amalan standard untuk memasang perpustakaan Python pada Debian 12 ialah memasangnya dalam persekitaran maya Python supaya ia tidak mengganggu pakej/pustaka Python sistem.
Untuk mencipta persekitaran maya Python 3 baharu untuk PyTorch dalam direktori “/opt/pytorch”, jalankan arahan berikut:
$ sudo python3 -m venv / memilih / pytorchMenaik taraf Python 3 PIP kepada Versi Terkini pada Persekitaran Maya Python 3 PyTorch
Untuk menaik taraf Python 3 PIP kepada versi terkini pada persekitaran maya '/opt/pytorch' Python 3, jalankan arahan berikut:
$ sudo / memilih / pytorch / tong sampah / pip3 pasang --naik taraf pip 
Memasang PyTorch dengan Sokongan Pecutan GPU/CUDA NVIDIA pada Debian 12
Untuk pecutan GPU/CUDA NVIDIA PyTorch berfungsi, anda mesti memasang versi PyTorch yang betul yang menyokong versi pemacu NVIDIA CUDA yang anda pasang pada sistem Debian 12 anda. Pada masa penulisan ini, PyTorch menyokong pemacu NVIDIA CUDA versi 11.8 dan 12.1. Untuk maklumat terkini tentang versi pemacu NVIDIA CUDA yang disokong PyTorch, semak laman web rasmi PyTorch .
Untuk menyemak versi pemacu NVIDIA CUDA yang anda pasang pada sistem Debian 12 anda, jalankan arahan berikut. Seperti yang anda lihat, kami telah memasang NVIDIA CUDA versi 11.8 pada sistem Debian 12 kami.
$ nvcc --versi 
Untuk memasang PyTorch dengan sokongan NVIDIA CUDA 11.8 pada persekitaran maya PyTorch Python 3, jalankan arahan berikut:
$ sudo / memilih / pytorch / tong sampah / pip3 pasang torchvision torchvision torchaudio --index-url https: // download.pytorch.org / whl / dengan 118Untuk memasang PyTorch dengan sokongan NVIDIA CUDA 12.1 pada persekitaran maya PyTorch Python 3, jalankan arahan berikut:
$ sudo / memilih / pytorch / tong sampah / pip3 pasang torchvision torchvision torchaudioPyTorch sedang dipasang pada persekitaran maya PyTorch Python 3. Ia mengambil sedikit masa untuk disiapkan.
Pada ketika ini, PyTorch harus dipasang pada persekitaran maya PyTorch Python 3
Mengaktifkan Persekitaran Maya PyTorch Python 3
Untuk mengaktifkan persekitaran maya '/opt/pytorch' PyTorch Python, jalankan arahan berikut:
$ . / memilih / pytorch / tong sampah / aktifkanPersekitaran maya PyTorch Python 3 harus diaktifkan.
Mengakses PyTorch dan Menyemak Jika NVIDIA GPU/CUDA Acceleration Tersedia
Untuk membuka shell interaktif Python 3, jalankan arahan berikut:
$ python3Cangkang interaktif Python 3 harus dibuka.
Pertama, import PyTorch dengan baris kod berikut:
$ obor import 
Untuk menyemak versi PyTorch yang anda pasang, jalankan baris kod berikut. Seperti yang anda lihat, kami menjalankan PyTorch 2.1.0 dengan sokongan pecutan NVIDIA CUDA 11.8 (cu118).
$ obor.__versi__ 
Untuk menyemak sama ada PyTorch mampu menggunakan GPU NVIDIA anda untuk pecutan NVIDIA CUDA, anda juga boleh menjalankan baris kod berikut. Jika sokongan NVIDIA CUDA tersedia, 'True' akan dicetak.
$ torch.cuda.is_available ( )Jika anda mempunyai berbilang GPU dipasang pada komputer anda, anda boleh menyemak bilangan GPU yang boleh digunakan oleh PyTorch dengan baris kod berikut. Seperti yang anda lihat, kami telah memasang NVIDIA GPU (RTX 4070) pada sistem Debian 12 kami.
$ torch.cuda.device_count ( )Untuk keluar daripada shell interaktif Python, jalankan baris kod berikut:
$ berhenti ( )Kesimpulan
Dalam artikel ini, kami menunjukkan kepada anda cara memasang persekitaran maya Python 3 PIP dan Python 3 (venv) pada Debian 12. Kami juga menunjukkan kepada anda cara mencipta persekitaran maya Python 3 untuk PyTorch pada Debian 12 dan cara memasang PyTorch dengan NVIDIA CUDA Sokongan pecutan 11.8 dan 12.1 pada Debian 12 juga. Akhir sekali, kami menunjukkan kepada anda cara mengaktifkan persekitaran maya PyTorch Python dan mengakses PyTorch pada Debian 12.