一、驱动:
下载地址:https://www.geforce.cn/drivers
安装驱动,选择自定义安装。
二、cuda安装:(tensorflow-gpu1.14.0使用的是10.0,所以做好安装10.0)
下载地址:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
- Double click cuda_10.1.243_426.00_win10.exe
10下载地址:https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
Double click cuda_10.0.130_411.31_win10.exe
打开之后,发现它自动多了CUDA_PATH的两个变量,接下来,我们还要添加:
CUDA_SDK_PATH = D:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0#这里是刚刚安装CUDA记录的路径
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64
CUDA_BIN_PATH = %CUDA_PATH%\bin
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64
D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
D:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0\common\lib\x64
D:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0\bin\win64
配置完成后,我们可以验证是否配置成功,主要使用CUDA内置的deviceQuery.exe 和 bandwithTest.exe:
首先win+R启动cmd,cd到安装目录下的 ...\extras\demo_suite
,然后分别执行bandwidthTest.exe
和deviceQuery.exe
,应该得到下图:
"%CUDA_PATH%\extras\demo_suite\bandwidthTest.exe"
"%CUDA_PATH%\extras\demo_suite\deviceQuery.exe"
三、cudnn安装:
从官网下载安装包:https://developer.nvidia.com/cudnn,注意需要注册为开发者。
安装完成之后,还需要下载cuDNN,这里需要登录并填写问卷才能下载,到下图所示的下载页面后,下载完成后,将这个压缩包里的所有文件放到CUDA10.1安装目录相应文件夹下即可。
需要将CUDNN解压后的3个文件夹拷贝到%CUDA_PATH%目录。否则接下来会遇到Could not find 'cudnn64_6.dll'.
按照自己的情况来,图中是我的解压和安装路径:(注意是路径中的文件)
添加环境变量:D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64
四、tensorflow-gpu(当前可用为1.14.0)
pip install tensorflow-gpu
import tensorflow as tf
sess = tf.Session()
a = tf.constant(1)
b = tf.constant(2)
print(sess.run(a+b))