Cara terbaru bisa mengikuti petunjuk disini https://www.tensorflow.org/install/pip#windows-native_1
Ini untuk versi Windows 10 (hanya untuk GPU Nvidia bukan utnuk AMD)
Cara menginstall tensorflow GPU pada laptop atau komputer yang memiliki GPU Nvidia tidak begitu susah.
Ada beberapa tahap untuk menjalankan tensorflow GPU di komputer atau laptop yang ada GPU-nya.
-
Install anaconda
-
Dari conda prompt sheel kita buat environment "conda create --name 'yourenv' python=3.6", contoh "conda create --name tf2-gpu python=3.6". Dalam perintah pembuatan environment tersebut kita menambahkan beberapa kode install, yaitu nama env (misalnya tf2-gpu) dan python versi 3.6 atau versi atas lainnya.
-
Setelah proses 2 selesai, lanjut install tensorflow-nya dengan mengetik kode pakai pip "pip install tensorflow-gpu==2.3.0", Note: tensorflow-gpu versi 2.0.0, 2.1.0, 2.2.0, dan 2.3.0 menggunakan CUDA 10 dan cuDNN 7.6, sedangkan versi tensorflow-gpu==2.4.0 menggunakan CUDA 11 dan cuDNN 8. Selengkapnya dapat dibaca di web tensorflow-nya langsung https://www.tensorflow.org/install/source_windows#gpu Cek tensorflow release di https://github.com/tensorflow/tensorflow/releases dan di https://github.com/tensorflow/tensorflow/releases dan disini juga https://www.tensorflow.org/versions#tensorflow_2
-
Buka python melalui Anaconda prompt shell lalu ketik "import tensorflow as tf", lalu enter
-
Jika ada error open dynamic cudart64_101.dll, download dahulu di link berikut https://www.dll-files.com/cudart64_101.dll.html, lalu extract dan simpan di C:// -> User -> NameUser -> Anaconda3 -> envs -> yourenv -> Libarry -> bin (paste disini)
-
Install cuDNN sesuai versi tensorflow, untuk tensorflow-gpu versi 2.3.0 pakai cuDNN versi 7.6 dan CUDA 10, ketik di anaconda prompt shell "conda install cudnn=7" lalu enter
-
Tes GPU dengan ketik "gpu_available = tf.test.is_gpu_available()" link: https://www.tensorflow.org/api_docs/python/tf/test/is_gpu_available
-
Jika tidak ada error lagi dan dynamic open berhasil, tinggal tes code GPU test di repo ini.
-
Link tutorial youtube https://youtu.be/G_HBDFXDXFI (Part 1) dan https://youtu.be/aWlDKT1NK1c (Part 2)
-
Link install lebih lengkap dapat dibaca di https://warstek.com/tensorflowgpu/
-
Happy Deep Learning
-
Test input
-
Jika gagal gunakan cara terbaru berikut: https://www.tensorflow.org/install/pip#windows-native