1.安裝Anaconda,添加環境變量;
anaconda官網 https://www.anaconda.com/distribution/
2.打開Anaconda navigator, 在environments下點擊Create,創建獨立的tensorflow環境;
3.在右上角搜索框中搜索tensorflow,選中搜索結果中的tensorflow-gpu版,點擊右下角的apply;
4.在設備管理器中核對電腦的GPU型號;
5.卸載電腦原有的顯卡驅動,在英偉達網站中選擇對應的GPU驅動,下載安裝;
https://www.nvidia.com/download/index.aspx?lang=en-us
6.在anaconda navigator - environment項目下,在創建的tensorflow環境中,搜索cud,
選中cudatoolkit 和 cudnn,點擊右下角的apply;
7.在IPython中測試;
import tensorflow as tf tf.test.gpu_device_name()
from tensorflow.python.client import device_lib device_lib.list_local_devices()
安裝過程中只有顯卡驅動是從英偉達官網下載,cuda toolkit 和 cudnn都是從anaconda中下載。
參考資料:https://cloud.tencent.com/developer/article/1165267
https://tensorflow.google.cn/install/gpu