版权声明:本文为博主原创文章,未经博主允许不得转载。 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
我用的这个链接的第六部分测试通过。