Bagaimana untuk memasang PyTorch di bawah Linux

1. Pengenalan kepada PyTorch

PyTorch ialah perpustakaan pembelajaran mesin Python sumber terbuka, berdasarkan Torch, digunakan untuk pemprosesan bahasa semula jadi dan aplikasi lain. Pada Januari 2017, PyTorch telah dilancarkan berdasarkan Torch oleh Facebook Artificial Intelligence Research Institute (FAIR). Pendahulu PyTorch ialah Torch Lapisan asasnya adalah sama dengan rangka kerja Torch, tetapi banyak kandungan telah ditulis semula dalam Python Ia bukan sahaja lebih fleksibel, menyokong grafik dinamik, tetapi juga menyediakan antara muka Python. Dibangunkan oleh pasukan Torch7, ia merupakan rangka kerja pembelajaran mendalam pertama Python yang bukan sahaja mendayakan pecutan GPU yang berkuasa, tetapi juga menyokong rangkaian saraf dinamik. PyTorch boleh dianggap sebagai numpy dengan sokongan GPU, dan ia juga boleh dianggap sebagai rangkaian saraf dalam yang berkuasa dengan fungsi terbitan automatik. Selain Facebook, ia telah diterima pakai oleh institusi seperti Twitter, CMU dan Salesforce.

2. Langkah-langkah pemasangan

1. Pemilihan sistem pengendalian

    , Kaedah pemasangan seperti pip dan kod sumber juga menyokong platform pengkomputeran CPU, cuda dan ROCm Apabila kita mengklik pada pemilihan persekitaran, kita boleh mendapati bahawa pada masa ini hanya sistem Linux menyokong semua bahasa, semua kaedah pemasangan dan semua platform pengkomputeran. jadi kami memilih sistem pengendalian Linux sebagai persekitaran sistem. Selain itu, pengiraan pembelajaran mesin memerlukan versi glibc yang lebih tinggi, versi kernel dan glbic centos adalah lebih rendah, dan versi kernel Ubuntu adalah lebih baharu, jadi adalah disyorkan untuk menggunakan sistem pengendalian Ubuntu untuk hos pembelajaran mesin. Pada masa ini, versi Ubuntu minimum yang disokong oleh kemas kini cuda ialah 18.04, jadi disyorkan untuk menggunakan sistem pengendalian Ubuntu 18.04 ke atas.

wuhs@s169:~$ cat /etc/os-release
NAME="Ubuntu"
VERSION="18.04.6 LTS ( Bionic Beaver)"

2. Pemasangan Anaconda3

   Seperti yang ditunjukkan di atas, PyTorch menyokong pelbagai kaedah pemasangan. Blogger merancang untuk menggunakan kaedah pemasangan conda. Ia adalah disyorkan untuk pertama Selepas memasang Anaconda3, kita boleh mencipta persekitaran maya yang berbeza mengikut keperluan kita Mesin yang berbeza versi PyTorch boleh dipasang dalam persekitaran maya saling menyokong tanpa menjejaskan satu sama lain. Untuk pemasangan anaconda dalam persekitaran Ubuntu, lihat catatan blog Pemasangan Anaconda3 Ubuntu.

wuhs@s169:~$ wget https://mirrors.bfsu.edu.cn/anaconda/archive/Anaconda3-2022.10-Linux-x86_64.sh
wuhs@s169:~$ sh Anaconda3-2022.10-Linux-x86_64.sh
wuhs@s169:~$ sumber ~/.bashrc

3. Semak versi Python

   Versi Python yang diperlukan oleh versi PyTorch adalah berbeza, jadi selepas memasang anaconda3, kami menyemak versi Python semasa Inisialisasi lalai ialah versi terkini Python yang sepadan dengan keluaran anaconda3 semasa, kami juga boleh menggunakan conda untuk mencipta yang diperlukan Versi persekitaran Python. Kami menyemak keperluan padanan versi PyTorch, torchvision dan Python di torchvision.

(asas) wuhs@s169:~$ python -V
Python 3.9.13

4

  Seperti yang ditunjukkan dalam langkah kedua, di tapak web rasmi PyTorch, kami boleh menjana arahan pemasangan yang sepadan selepas memilih sistem pengendalian, kaedah pemasangan, bahasa pengaturcaraan dan platform pengkomputeran.

(asas) wuhs@s169:~$ conda install pytorch torchvision torchaudio cpuonly -c pytorch
Mengumpul metadata pakej (current_repodata.json): selesai
Persekitaran penyelesaian: selesai

## Pelan Pakej ##

Teruskan ([y]/n) y
…

5. Pengesahan versi

(asas) wuhs@s169:~$ python
Python 3.9.13 (utama, 25 Ogos 2022, 23:26:10)
[GCC 11.2 .0] :: Anaconda, Inc. di linux
Taip “bantuan”, “hak cipta”, “kredit” atau “lesen” untuk mendapatkan maklumat lanjut.
>>> > obor. >1. Cipta persekitaran maya
(asas) wuhs@s169:~$ conda create -n pytorch python=3.9
…
(base) wuhs@s169 :~$ conda activate pytorch

(pytorch) wuhs@s169:~$

2 Pasang versi PyTorch yang ditentukan

  Apabila memasang versi yang ditentukan. daripada PyTorch, kami perlu menyemak versi yang sepadan pada saluran PyTorch pada laman web rasmi GitHub dan TorchAudio, lihat

TorchAudio. Sudah tentu, jika kami menyatakan versi yang salah, ralat akan dilaporkan semasa pemasangan Kami boleh menyemak versi perisian yang salah mengikut mesej ralat, kemudian pergi ke laman web rasmi untuk mengesahkan pembetulan dan memasang semula.

(pytorch) wuhs@s169:~$ conda install pytorch2.12.0 torchvision=0.13.0 torchaudio0.12.0 cpuonly -c pytorch

Atas ialah kandungan terperinci Bagaimana untuk memasang PyTorch di bawah Linux. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!