Install TensorFlow from Source with GPU Support

  • Posted on: 17 January 2017
  • By: oon
Install TensorFlow from Source with GPU Support

Meski banyak sekali cara instalasi TensorFlow [1], ternyata lebih banyak lagi artikel di internet yang menyarankan untuk install sendiri TensorFlow dari source jika menggunakan komputer yang mendukung GPU.

Salah satu sumber utama yang digunakan acuan adalah referensi dari situs Nvidia [2].
Atau contoh lain yang tulisannya lebih detil (dan sesuai kebutuhan penulis untuk GPU 1080) adalah artikel marcnu [3].

Kedua artikel di atas kurang lebih menggambarkan proses instalasi dengan syarat Nvidia GPU sudah terdeteksi di sistem operasinya, misal di Ubuntu bisa mengacu ke artikel "Proprietary (Nvidia) GPU Drivers by Graphics Drivers Team" agar proses instalasi driver Nvidia tidak perlu secara manual (dari file run).

Bagian yang perlu diperhatikan adalah setelah terinstall CUDA, kita perlu install cuDNN (library dengan dukungan GPU untuk Deep Neural Network). Nah untuk install cuDNN ini kita harus buat account dahulu di situsnya Nvidia serta mengisi sejumlah survei yang berhubungan dengan untuk apa kita mau download library tersebut.

Untuk versi TensorFlow yang mau diinstal dari source bisa dipilih dari https://www.tensorflow.org/versions/

Setelah semua proses instalasi berakhir, kita bisa membuat file hello world sekaligus test, dengan hasil:

Test TensorFlow Installation with GPU Support

[1]https://www.tensorflow.org/get_started/os_setup
[2]http://www.nvidia.com/object/gpu-accelerated-applications-tensorflow-ins...
[3]https://marcnu.github.io/2016-08-17/Tensorflow-v0.10-installed-from-scra...
[4]https://github.com/floydhub/dl-setup

 

Add new comment

Filtered HTML

  • Web page addresses and e-mail addresses turn into links automatically.
  • Allowed HTML tags: <a> <em> <strong> <cite> <blockquote> <code> <ul> <ol> <li> <dl> <dt> <dd>
  • Lines and paragraphs break automatically.

Plain text

  • No HTML tags allowed.
  • Web page addresses and e-mail addresses turn into links automatically.
  • Lines and paragraphs break automatically.
By submitting this form, you accept the Mollom privacy policy.