软件版本:
1, tensorflow : 1.2.1-gpu
- cuda : 8.0.
3, cudnn: 5.1 for cuda 8.0
4, python : 3.5
5, pip 9.0.1
注意事项:
1,在window平台上,只支持python3.5以上版本(推荐使用3.5),使用3.6会有一些坑
2, cuda 和 cudnn 的版本也要对应
3,cudnn这个库建议按照以下顺序加入到对应文件夹:
cudnn64_5.dll -----------------CUDA\v8.0\bin
cudnn.h------------------------CUDA\v8.0\include
cudnn.lib-----------------------CUDA\v8.0\lib\x64
网友评论