tensorflow gpu di ubuntu
Machine Learning

Cara Install Tensorflow GPU di Ubuntu 20.04

Ubuntu adalah sistem operasi open source yang berbasiskan Debian. Ubuntu menawarkan 3 edisi yang dapat digunakan yaitu Ubuntu Desktop, Ubuntu Server dan Ubuntu Core

Distro Ubuntu cukup mudah digunakan karena cukup nyaman dan menyerupai Windows

Beberapa hari lalu saya berkesempatan menginstall Ubuntu Desktop terbaru yaitu Ubuntu 20.04 LTS dengan nama resminya adalah Focal Fossa

Ubuntu mempunyai support panjang yang disebut dengan LTS (Long Term Service)

LTS memungkinkan kita untuk stand by di distro linux untuk waktu yang lama karena ada support untuk distro tersebut

Ubuntu mengeluarkan versi LTS setiap 2 tahun sekali. Berikut adalah rilis Ubuntu LTS beberapa tahun terakhir beserta namanya

  1. Ubuntu 14.04 LTS -> Trusty Tahr
  2. Ubuntu 16.04 LTS -> Xenial Xerus
  3. Ubuntu 18.04 LTS -> Bionic Beaver
  4. Ubuntu 20.04 LTS -> Focal Fossa

eniwei, setelah beberapa hari ini menggunakan Ubuntu 20.04 LTS saya merasa cukup puas baik dari performance maupun dari sisi tampilan yang lebih elegen dibandingkan dengan versi-versi sebelumnya

Ubuntu 20.04 LTS Focal Fossa

Install Tensorflow GPU

Oke selanjutnya kita akan coba install Tensorflow GPU di Ubuntu 20.04

Pertama kita kenalan dulu dengan Tensorflow

Tensorflow adalah library yang dikembangkan oleh Google untuk mengembangkan aplikasi deep learning. Tensorflow dapat menggunakan CPU dan GPU untuk memproses datanya

Untuk membangun aplikasi neural network baiknya gunakan GPU untuk melakukan proses training karena akan lebih cepat dibandingkan dengan CPU

Berbandingan antara GPU dan CPU untuk Deep Learning dapat dibaca di artikel ini : Do We Really Need GPU for Deep Learning?

Untuk menggunakan Tensorflow GPU kita harus install CUDA dan cuDNN terlebih dahulu

Referensi yang saya gunakan dapat dilihat di installing tensorflow gpu in ubuntu 20.04

Install CUDA

Pertama pastikan bahwa laptop yang anda gunakan support untuk GPU

Kemudian kita install nvdia-cuda-toolkit dengan perintah

sudo apt install nvidia-cuda-toolkit

Setelah proses instalasi selesai, Jalankan sintak nvcc -V di terminal untuk melakukan verfikasi instalasi telah berhasil

untuk melihat lokasi CUDA terinstall gunakan sintak whereis cuda

Install cuDNN

Selanjutnya kita akan install cuDNN

Karena kita menggunakan CUDA versi 10.1 maka kita cari terlebih dahulu versi cuDNN yang support dengan CUDA versi 10.1 di halaman resmi NVDIA

Silahkan mengunjungi web official NVDIA untuk mendownload package cuDNN

Biasanya diawal akan diminta untuk login, jika belum punya akunnya silahkan register terlebih dahulu

Archive cuDNN repository

Karena kita menggunakan CUDA 10.1 kita bisa download cuDNN versi 7.6.5

kemudian download cuDNN Library for Linux

File yang kita download adalah cudnn-10.1-linux-x64-v7.6.5.32.tgz

Selanjutnya kita ekstrak dan kita pindahkan ke direktori CUDA

tar -xvzf cudnn-10.1-linux-x64-v7.6.5.32.tgz
sudo cp cuda/include/cudnn.h /usr/lib/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/lib/cuda/lib64/

Kemudian kita set permissionnya

sudo chmod a+r /usr/lib/cuda/include/cudnn.h /usr/lib/cuda/lib64/libcudnn*

Kemudian export CUDA environment variable-nya

echo 'export LD_LIBRARY_PATH=/usr/lib/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/lib/cuda/include:$LD_LIBRARY_PATH' >> ~/.bashrc

Jalankan perintah

source ~/.bashrc

Install Tensorflow

Setelah CUDA dan cuDNN telah sukses diinstall saat kita instal library Tensorflow dan mengecek apakah Tensorflow sudah menggunakan GPU

Install Tensorflow

pip install tensorflow

Kemudian lakukan verifikasi bahwa Tensorflow sudah menggunakan GPU dengan masuk ke environment Python dan ketik perintah seperti ini

>>> import tensorflow as tf
>>> tf.config.list_physical_devices("GPU")

Di akhir proses kita mendapatkan informasi adalah [PhysicalDevice(name=’/physical_device:GPU:0′, device_type=’GPU’)]

atau kita juga bisa melakukan verifikasi dengan perintah seperti ini

>>> tf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None)

Diperlukan beberapa beberapa detik untuk mendapatkan hasilnya yang diakhir ditandai dengan kembalian True

Di akhir output kita mendapatkan pesan bahwa physical GPU telah terdeteksi dengan tipe GeForce MX250 dan compute capability: 6.1 (tipe GPU laptop saya)

2020-09-13 07:48:03.574153: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1402] Created TensorFlow device (/device:GPU:0 with 1311 MB memory) -> physical GPU (device: 0, name: GeForce MX250, pci bus id: 0000:02:00.0, compute capability: 6.1)
True

Oke sekian tutorial instalasi Tensorflow GPU di Ubuntu 20.04, selamat mencoba 😀

Leave a Reply

Your email address will not be published. Required fields are marked *