1. 程式人生 > >win10下安裝tensorflow-gpu+anaconda+pycharm

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的版本