机器学习环境搭建: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则配置成功。

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值