环境
- WIN 10 + CUDA 10.0 + cuDNN 7.4 + Anaconda3 + Tensorflow-gpu 1.13
遇到的坑
萌新一枚,有什么没有说对的地方还请指正 T^T
CUDA、Anaconda的安装过程就不详细说明,网上基本都全的,这里主要罗列了几个注意点和坑 :
- cuDNN和CUDA要匹配,写在cuDNN文件名里了
- Anaconda里面创建tensorflow-gpu环境的时候记得指定一下python版本,之后tf的轮子要匹配(cp36就是python3.6)
- 因为作了个死下了CUDA 10,和tf 1.12匹配不了不知道为什么(但是看到有博主成功过,感觉稳妥一点还是CUDA 9),后来换了1.13就OK了,来来去去费了挺大劲
- 下载轮子时,pip也要添加一下清华源,不然默认路径就龟速:
https://pypi.tuna.tsinghua.edu.cn/simple