Conda ile Pytorch İçin Ubuntu 16.04 GPU Kurulumu

Evet, CUDA, CuDnn, Nvidia sürücüleri ve Pytorch’u yüklemek baş belası olabilir. Ancak, Anaconda’nın yardımıyla çok kolay! LD_LIBRARY_PATH gibi ortam değişkenleriyle uğraşmanıza bile gerek yok. Docker kullanmanın daha kolay ve basit olabileceğini anlıyorum. Ancak henüz liman işçileri ile tanışmadım ve ortamlarımı Anaconda ile yönetmeye alışkınım. İşte burada!

Yalnızca İki Adım:

– Nvidia sürücüsünü yükleyin

– Conda yükleme paketleri

1. Nvidia Sürücüsünü Yükleyin

GPU’nuzu hangi sürücünün desteklediğini buradan kontrol edin. Normalde 3 basamaklı bir sayıdır. Örneğin, 440 şu anda en son sürümdür (10 Nisan 2020). Ancak, en son sürümün kararlı olmaması ihtimaline karşı 410’u yüklemeyi seçtim.

Aşağıdaki tabloya bakarak hangi sürücüyü ve hangi CUDA’yı yüklemek istediğinizi düşünün. Benim durumumda CUDA 9.2 ve Nvidia-410’u seçtim. Kesinlikle CUDA 10.0’ı da seçebilirim, ancak bu gerçekten size kalmış. Bazen bir depodaki bazı kodları çoğaltmak isterseniz, yazar tarafından önerilen bir CUDA sürümü olacaktır.

Yazılımlar & amp; Güncellemeler , diğer yazılımlar sekmesine tıklayın, grafik sürücüleri ppa hakkında bir şeyler yazan kutunun işaretini kaldırın. Bu adım, sürücünün otomatik olarak güncellenmemesini ve diğer şeyleri bozmamasını sağlar.

Nvidia sürücü sürümünüzü kontrol etmek için:

Sürüm, çıktıların en üstünde görünmelidir. Benim durumumda 410.78.

2. Conda Yükleme Paketleri

Anaconda’nın zaten kurulu olduğunu varsayıyorum (muhtemelen ilgili bir şeyi arayarak buraya gelmenizin nedeni). Ancak, durum bu değilse, bu eğiticiyi izleyerek anaconda’yı yükleyin.

Burada istediğiniz platform kombinasyonlarını seçerek conda install komutunu kontrol edin. Normalde, çok eski sürümlerden bazılarını yüklemeyi planlamıyorsanız, bu nispeten basittir.

Örneğin, CUDA 9.2’yi yüklemek istersem şu komutu kullanırım:

Pytorch web sayfasında, önerilen komut cudnn içermez, ancak ihtiyacımız olan her şeyi tek bir satıra kurmanın ve bizim için paket uyumluluk sorunlarını çözmesine izin vermenin daha uygun olduğunu düşünüyorum.

İşte bu! CUDA ve CuDnn’yi kaynaktan veya Nvidia.com’dan dosya indirerek yüklemenize gerek yoktur. Tek yapmanız gereken conda install .

Son olarak, paketlerin doğru şekilde kurulup kurulmadığını kontrol etmek için:

Hepsi bu kadar! Bunu yararlı bulursanız, lütfen beni alkışlayın.

Yanlış veya belirsiz bir şey varsa, lütfen yanıt verin.