win10系统tensorFlow-GPU版安装简记

本文记录了在神舟Z7-KP7EC笔记本上安装配置TensorFlow-GPU、CUDA及cuDNN的过程,包括解决显卡驱动兼容性问题及测试运行。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

环境:

神舟Z7-KP7EC+win10+gtx1060

过程:

1. 探索CUDA版本与显卡型号是否支持,tf版本、CUDA与CUDNN对应关系的问题

https://blog.youkuaiyun.com/XCCCCZ/article/details/80385448

2.安装tensorflow-gpu、cuda、cudnn

这个过程就千奇百怪了,不同的贴子有不同的现象,下面是我的:
  我没搞懂vs是不是必要,也没像大多数帖子那样用conda,就只装了python3.6.5(用virtualenv管理环境我觉得很方便),最新版tf是1.8.0,所以它需要的库如下:

    msvcp_dll_name = 'msvcp140.dll'
    cudart_dll_name = 'cudart64_90.dll'
    cuda_version_number = '9.0'
    nvcuda_dll_name = 'nvcuda.dll'
    cudnn_dll_name = 'cudnn64_7.dll'
    cudnn_version_number = '7'

其中msvcp140.dll和nvcuda.dll在system32下面都有(没装vs),意思就只差cudart64_90.dll(cuda9.0)和cudnn64_7.dll(cudnn7的bin目录下)了。
  安装cuda在硬件检测就出了警告,提示我没有找到相应的图显硬件,百度到cuda是公版驱动,而笔记本显卡一般是非公版,所以它识别不到,尝试了好久的添加“设备实例路径”到某个.inf文件里失败了,连那个.inf文件都没找到。最后决定硬着头皮忽略那个警告,安照警告内容只勾选cuda下的只勾选CUDA里的Development、Runtime、Documentation、Samples,不要勾选Visual Studio Integration。安装完后解压cudnn,只拷贝其下的文件到cuda安装路径下的对应文件夹下,没有出现覆盖的情况。另:我没安装cuda的补丁。

3.测试

https://blog.youkuaiyun.com/weixin_42359147/article/details/80622306
我用的这个链接的第六部分测试通过。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值