TensorFlow GPU 安裝的問題處理
電腦配置
- win10 簡體中文版
- gtx1060 顯卡
過程
anaconda 安裝
- 先安裝anaconda集成了一些必要工具，比全部自己折騰好點。
不過還是踩到坑 【https://www.cnblogs.com/bikaqiou2000/p/11827881.html】 總結了踩坑的經歷 - 完了后就安裝好了 jupyter,conda 等必要工具
安裝tensorflow cpu版
- 最初用pip安裝tensorflow，各種問題，不行 不再細說
- 用conda 安裝 cup版：如果只是安裝cpu版 一般
conda install tensorflow
基本會成功安裝最新版，不過用起來會慢很多
安裝tensorflow gpu版
- 我選擇安裝gpu版
- 先看官方文檔【https://tensorflow.google.cn/install/pip?lang=python3】，【https://tensorflow.google.cn/install/gpu】 里面說的必要條件要准備
- 開始安裝，沒有意外則可以成功
conda install tensorflow-gpu
- 安裝要加速可以看看 這個【https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/】
- 然后運行下官方的代碼
python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
顯然要翻車
安裝nvidia相關工具
-
原因是NVIDIA 的相關工具還沒有安裝 接着看
-
下載並安裝cuda 【https://developer.nvidia.com/cuda-toolkit】
-
下載cudnn 【https://developer.nvidia.com/rdp/cudnn-download】 中間官方網站要你注冊和填寫問卷，你就照做吧，不要錢的...
-
注意cuda toolkit 和 cudnn 版本一直性 10.1 cuda 就用 10.1 專屬的 cudnn
-
下載cudnn 完了解壓后將里面 三個文件夾 【bin，include，lib】放入cuda的安裝目錄。比如：【C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1】
-
然后再次運行下官方的代碼
python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
還是翻車，錯誤提示【CUDA driver version is insufficient for CUDA runtime version】
- 繼續處理問題參考【https://www.jianshu.com/p/f85c30650028】 上面說我的顯卡驅動版本太老了，更新前是 391.25，所以要更新最新驅動
C:\Program Files\NVIDIA Corporation\NVSMI> nvidia-smi
支持不到cuda 10
- 繼續去官網 【https://www.nvidia.cn/Download/index.aspx?lang=cn#】 下載你的顯卡對應驅動版本，我的gtx1060下載了441.2 可以支持到cuda 10.2
- 一路安裝完最新驅動，再看看 【nvidia-smi】命令提示的顯卡信息不是變化了
- 完了之后 執行 上面的tensorflow官方測試代碼估計能成功了
...
tf.Tensor(-1986.1196, shape=(), dtype=float32)