Win10安裝CUDA 9.0+cuDNN v7.0+tensorflow-gpu 1.11.0和opencv的教程

鑑於我自己踩過的坑,提醒大家千萬不要直接pip install tensorflow-gpu而不指定版本!!!我一開始用最新的1.13.1版本總是報“ImportError: DLL load failed: 找不到指定的模塊。”的錯誤,果斷的卸載後安裝了1.11.0版本,直接就OK啦!

由於我之前已經安裝了vs2015、Anaconda和Pycharm,就不在此贅述了。

本機配置:

  • GTX 1060 3G       
  • Win10 64              
  • vs2015                    
  • Anaconda3 5.1.0  
  • Pycharm                  

安裝GPU的TensorFlow必須先安裝CUDA和cuDNN,而且這三者之間的版本對應必須格外注意!!!

注:CUDA是顯卡驅動程序,cuDNN是用來加速深度學習訓練的庫

我自己採用的是CUDA 9.0+cuDNN v7.0+tensorflow-gpu 1.11.0

一、安裝CUDA

CUDA版本:cuda_9.0.176_win10(配套的cuDNN是V7.0)

下載地址:https://developer.nvidia.com/cuda-90-download-archive

CUDA的安裝十分簡單,根據自己的系統選對應的下載包,使用默認的精簡安裝,根據提示一直下一步就可以了。

安裝好後,在命令行輸入 nvcc -V,顯示如下即可。

二、安裝cuDNN版本:cudnn-9.0-windows10-x64-v7

下載地址:https://developer.nvidia.com/cudnn
下載解壓後文件如下:

直接將上面顯示的所有文件複製,粘貼到安裝的CUDA的路徑下即可,同名的文件夾會自動合併。

通常的默認路徑是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

三、配置環境變量

一般情況在安裝CUDA的時候下面兩張圖是配好的。

我們需要手動配置的環境變量如下:

四、安裝tensorflow-gpu

1.在Anaconda新建一個tensorflow-gpu環境(當然不想新建環境也可以直接在Anaconda的root環境下安裝)

conda create -n tensorflow-gpu python=3.6

(D:\Python\Anaconda3-5.0.1) C:\Users\sxq>conda create -n tensorflow-gpu python=3.6
Fetching package metadata .....
Solving package specifications: .

Package plan for installation in environment D:\Python\Anaconda3-5.0.1\envs\tensorflow-gpu:

The following NEW packages will be INSTALLED:

    certifi:        2016.2.28-py36_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
    pip:            9.0.1-py36_1     https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
    python:         3.6.2-0          https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
    setuptools:     36.4.0-py36_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
    vc:             14-0             https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
    vs2015_runtime: 14.0.25420-0     https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
    wheel:          0.29.0-py36_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
    wincertstore:   0.2-py36_0       https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free

Proceed ([y]/n)? y

#
# To activate this environment, use:
# > activate tensorflow-gpu
#
# To deactivate an active environment, use:
# > deactivate
#
# * for power-users using bash, you must source
#

2.建立tensorflow環境後,每次需要在該環境下工作,我們需要先激活tensorflow-gpu進入環境

activate tensorflow-gpu(退出是deactivate tensorflow-gpu)

3.安裝所需的包

gpu版本的tensorflo輸入以下命令:

pip install --ignore-installed --upgrade tensorflow-gpu==1.11.0

更新pip命令:

python -m pip install --upgrade pip

可用如下代碼測試tensorflow-gpu是否安裝成功:

import tensorflow as tf

# 新建一個 graph.
with tf.device('/gpu:0'):
    a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
    b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
    c = tf.matmul(a, b)
# 新建 session with log_device_placement 並設置爲 True.
sess = tf.Session(config=tf.ConfigProto(
    allow_soft_placement=True, log_device_placement=True))
# 運行這個 op.
print(sess.run(c))

五、安裝OpenCV

 pip install opencv-python

可用如下代碼測試OpenCV是否安裝成功:

import cv2

img = cv2.imread("D:\dog.jpg")
cv2.imshow("dog", img)
cv2.waitKey()

 

所有安裝完成後,使用conda list查看已安裝的包如下所示:

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章