win10下安裝tensorflow-gpu+anaconda+pycharm
電腦配置:i5+gtx1050
需要的材料:cudnn-9.0-windows10-x64-v7.1 + Anaconda3-5.2.0-Windows-x86_64 + cuda_9.0.176_win10 + JetBrains PyCharm Community Edition 2018.1.4 x64
國內安裝源:pip install --ignore-installed --upgrade tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple
ps:這兩個的選擇要根據tensorflow版本選擇
Anaconda3-5.2.0-Windows-x86_64和PyCharm 在官網下載,PyCharm 我選的Community版,配置python時都一樣。
安裝順序無所謂,關鍵是cuda,cudnn版本要對。
Anaconda,PyCharm,cuda安裝一直點下一步就行。
解壓cudnn,解壓厚的資料夾裡的檔案複製到 .....\NVIDIA GPU Computing Toolkit\CUDA\v9.0\目錄下
將F:\ANACONDA和F:\ANACONDA\Scripts新增到環境變數中
開啟Windows power shell 管理員 輸入pip install --ignore-installed --upgrade tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple
安裝完成後,輸入:pip list
能找到tensorflow-gpu和tensorboard 說明安裝成功
在pycharm中操作如圖
確定設定即可
下面是安裝tennsorflow-gpu後常見問題:
FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.from ._conv import register_converters as _register_converters
解決方法:升級元件:pip install h5py==2.8.0rc1
檢測是否使用gpu:
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
Loaded runtime CuDNN library: 7104 (compatibility version 7100) but source w........
沒有安裝對應版本的cudnn
解決方法:跟換cudnn的版本或tensorflow的版本