折腾一晚上,主要是版本没对上,做个笔记。
1、安装Anaconda3
官网下载后安装,我用的5.2.0版本。
这个没问题,需要注意的是环境变量的设置选项,勾选下,就不用自己费劲了。(我第一次自己设环境变量,一直不可以,后来发现是在用户变量中添加,而不是系统变量中,郁闷)
2、安装CUDA9.0 + cudnn7
CUDA在官网下载,根据自己电脑选择。
cudnn7同理。
这里需要注意额是环境变量的设置、版本对应问题。(踩坑:我电脑版本支持CUDA9.1,我就直接下载安装,折腾半天后发现,9.1版本不支持tensorflow,气人,卸载重装9.0。也就是说安装软件前要做好调研啊,不要一味追求最新版本)
3、安装tensorflow
刚开始是1.10.0,跟CUDA9.0不支持,没办法那就降级吧,最后安装的1.6.0的GPU版本。
其中有个小插曲(import导入成功,但会有np警告,此处降级numpy版本即可):https://blog.csdn.net/ericcchen/article/details/79550472
当然,也可以用conda开个虚拟环境安装,一样样的。
总结:
tensorflow_gpu版本安装CUDA+cudnn,要好好注意版本问题,尽量避免踩坑。哈哈,其实坑踩实了,以后再装也不怕啦。