win10系统tensorFlow-GPU版安装简记

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/x15696576570/article/details/80787676

环境:

神舟Z7-KP7EC+win10+gtx1060

过程:

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

https://blog.csdn.net/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.csdn.net/weixin_42359147/article/details/80622306
我用的这个链接的第六部分测试通过。

猜你喜欢

转载自blog.csdn.net/x15696576570/article/details/80787676