Bagaimana sokongan GPU untuk Pytorch di CentOS

Untuk membolehkan pecutan GPU pytorch pada sistem CentOS, anda perlu memasang versi CUDA, CUDNN dan GPU PYTORCH. Langkah -langkah berikut akan membimbing anda melalui proses:

Pemasangan CUDA dan CUDNN

1. Sahkan keserasian versi CUDA: Gunakan perintah nvidia-smi untuk melihat versi CUDA yang disokong oleh kad grafik NVIDIA anda. Sebagai contoh, kad grafik MX450 anda boleh menyokong CUDA versi 11.1 atau lebih baru.

2. Muat turun dan pasang CUDA Toolkit: Lawati laman web rasmi NVIDIA CUDA Toolkit dan muat turun dan pasang versi yang sepadan mengikut versi CUDA tertinggi yang disokong oleh kad grafik anda.

3. Pasang Perpustakaan CUDNN: Pergi ke laman web rasmi NVIDIA CUDNN , muat turun perpustakaan CUDNN yang serasi dengan versi CUDA anda, dan ikuti panduan rasmi untuk menyelesaikan pemasangan.

Pemasangan versi GPU pytorch

1. Pasang versi GPU PyTorch dengan PIP: Gunakan arahan PIP untuk memasang versi GPU PyTorch yang serasi mengikut versi CUDA anda. Sebagai contoh, untuk CUDA 11.1, anda boleh merujuk kepada arahan yang disediakan oleh laman web rasmi PyTorch untuk memasangnya, dan pastikan untuk memilih versi yang sepadan dengan versi CUDA dan CUDNN anda.

Sahkan sokongan GPU

1. Semak ketersediaan CUDA: Gunakan kod python berikut untuk mengesahkan bahawa CUDA dipasang dengan betul dan tersedia:

    obor import
   cetak (obor.cuda.is_available ()) # output benar bermakna cuda tersedia

2. Dapatkan Maklumat GPU: Jalankan kod berikut untuk mendapatkan bilangan GPU, nombor peranti GPU yang sedang digunakan, dan nama GPU:

    cetak (obor.cuda.device_count ()) # output bilangan cetak gpus (obor.cuda.current_device ()) # output nombor peranti gpu semasa cetak (obor.cuda.get_device_name (0)) # output nama peranti gpu pertama

Sekiranya langkah -langkah di atas berjaya diselesaikan, anda boleh menggunakan ciri pecutan GPU PyTorch pada sistem CentOS anda. Jika anda mempunyai sebarang pertanyaan, sila rujuk dokumentasi Pytorch rasmi atau forum komuniti yang berkaitan untuk mendapatkan bantuan.

Atas ialah kandungan terperinci Bagaimana sokongan GPU untuk Pytorch di CentOS. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!