机器学习环境搭建:Win8.1+Python3.6+TensorFlow-GPU1.5.0+CUDA9.0+cudnn7

学习机器学习,我选择了Google的TensorFlow,为了充分利用资源,决定安装GPU版本,那么理所当然的要安装好英伟达显卡的计算平台CUDA+cudnn。

TF1.5.0版本已经支持CUDA9.0+cudnn7,具体下载以及安装不再详细讲述自行百度,这里讲一下遇到的问题以及解决方法。(cudnn官网没有win8的版本,用win7的也行)

安装TF没有什么问题,直接pip安装就可以,最大的问题在与CUDA9.0的环境设置。下载好cudnn后,将里面的三个文件夹解压加入到cuda里面,然后设置系统变量:(有就不管,没有就添加,要修改就修改)

CUDA_BIN_PATH:%CUDA_PATH%\bin

CUDA_LIB_PATH:%CUDA_PATH%\lib\x64

CUDA_PATH:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

CUDA_PATH_V9_0:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

CUDA_SDK_BIN_PATH:C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0\bin

CUDA_SDK_LIB_PATH:C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0\common\lib\x64

CUDA_SDK_PATH:C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0

最后在path中添加:%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64;C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0\bin\win64;C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0\common\lib\x64 


注意:win8.1的C盘里的ProgramData这个文件夹是默认隐藏文件夹。


添加修改完成后重启电脑。找到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite文件夹,用cmd运行该目录下的bandwidthTest.exe和deviceQuery.exe,如果都显示pass则配置成功。

猜你喜欢

转载自blog.csdn.net/LawGeorge/article/details/79226794