win10+cuda9.0+cuDNN 7.0+TensorFlow-gpu install steps

 Official Tutorials:

4. Installing cuDNN on Windows

4.1. Prerequisites

Ensure you meet the following requirements before you install cuDNN.

  • A GPU of compute capability 3.0 or higher. To understand the compute capability of the GPU on your system, see: CUDA GPUs.
  • One of the following supported platforms:
    • Windows 7
    • Windows 10
    • Windows Server 2012
  • One of the following supported CUDA versions and NVIDIA graphics driver:
    • NVIDIA graphics driver R377 or newer for CUDA 8
    • NVIDIA graphics driver R384 or newer for CUDA 9
    • NVIDIA graphics driver R390 or newer for CUDA 9.2
    For more information, see

4.1.1. Installing NVIDIA Graphics Drivers

Install up-to-date NVIDIA graphics drivers on your Windows system.

  1. Go to: NVIDIA download drivers
  2. Select the GPU and OS version from the drop down menus.
  3. Download and install NVIDIA driver as indicated in that webpage. For more information, select the ADDITIONAL INFORMATION tab for step-by-step instructions for installing a driver.
  4. Restart your system to ensure the graphics driver takes effect.

4.1.2. Installing CUDA

Refer to the following instructions for installing CUDA on Windows, including the CUDA driver and toolkit: NVIDIA CUDA Installation Guide for Windows.

4.2. Downloading cuDNN

In order to download cuDNN, ensure you are registered for the NVIDIA Developer Program.

  1. Go to: NVIDIA cuDNN home page.
  2. Click Download.
  3. Complete the short survey and click Submit.
  4. Accept the Terms and Conditions. A list of available download versions of cuDNN displays.
  5. Select the cuDNN version to want to install. A list of available resources displays.
  6. Extract the cuDNN archive to a directory of your choice.

4.3. Installing cuDNN on Windows

The following steps describe how to build a cuDNN dependent program. In the following sections:

  • your CUDA directory path is referred to as C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
  • your cuDNN directory path is referred to as <installpath>

  1. Navigate to your <installpath> directory containing cuDNN.
  2. Unzip the cuDNN package.
    cudnn-9.0-windows7-x64-v7.zip
    or
    cudnn-9.0-windows10-x64-v7.zip
  3. Copy the following files into the CUDA Toolkit directory.
    1. Copy <installpath>\cuda\bin\cudnn64_7.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin.
    2. Copy <installpath>\cuda\ include\cudnn.h to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include.
    3. Copy <installpath>\cuda\lib\x64\cudnn.lib to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64.
  4. Set the following environment variables to point to where cuDNN is located. To access the value of the $(CUDA_PATH) environment variable, perform the following steps:
    1. Open a command prompt from the Start menu.
    2. Type Run and hit Enter.
    3. Issue the control sysdm.cpl command.
    4. Select the Advanced tab at the top of the window.
    5. Click Environment Variables at the bottom of the window.
    6. Ensure the following values are set:
      Variable Name: CUDA_PATH 
      Variable Value: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
      
  5. Include cudnn.lib in your Visual Studio project.
    1. Open the Visual Studio project and right-click on the project name.
    2. Click Linker > Input > Additional Dependencies.
    3. Add cudnn.lib and click OK.

4.4. Upgrading from v6 to v7

cuDNN v7 can coexist with previous versions of cuDNN, such as v5 or v6.

4.5. Troubleshooting

Join the NVIDIA Developer Forum to post questions and follow discussions.

 

Install and Test Tensorflow Using GPU:

If you have installed the Cpu version of Tensorflow,use the command:pip uninstall tensorflow to uninstall it

then using the command:pip install tensorflow-gpu 

这里注意安装的tensorflow版本一定要和cuda-cudnn匹配,不然会报错!

版本信息请参阅这篇文章:https://blog.csdn.net/qq_36396104/article/details/89165185

Test tensorflow,use this code

import sys
import numpy as np
import tensorflow as tf
from datetime import datetime

### argv[1] = type of device and which one
### argv[2] = size of the matrix to operate on

device_name = sys.argv[1]
shape = (int(sys.argv[2]), int(sys.argv[2]))
if device_name == "gpu":
    device_name = "/gpu:0"
else:
    device_name = "/cpu:0"

with tf.device(device_name):
    random_matrix = tf.random_uniform(shape=shape, minval=0, maxval=1)
    dot_operation = tf.matmul(random_matrix, tf.transpose(random_matrix))
    sum_operation = tf.reduce_sum(dot_operation)

startTime = datetime.now()
with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as session:
        result = session.run(sum_operation)
        print(result)

### Print the shape, device name and timing
print("\n" * 3)
print("Shape:", shape, "Device:", device_name)
print("Time taken:", datetime.now() - startTime)
print("\n" * 3)

then run the  script in the command window:

python tensorflow_test.py gpu 10000

if you successfully configured the environment,youwill see this:

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值