问题描述
由于深度学习的需要,所以需要安装tensorflow-gpu来进行学习,虽然以前在别的电脑上以及服务器上都安装过,搭建过,但是这次在自己新电脑上还是踩了雷区,折磨了半天,但是最后发现是我傻了!!!!!除了传统的方法还有一种简单的方法。
在描述前,先放一个驱动版本和CUDA的对应表,不对应的话肯定是错的!!!!
方法描述
传统方法安装
1.安装tensorflow-gpu 1.9
这里推荐使用镜像,那速度快到爆炸!!!
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.9
2.安装CUDA和cudaa
这里推荐一个博客,里面有相对较详细的教程 l链接点我.
注意注意!!!这里就要说坑了
1.一般电脑要安装VS2015,因为里面有一个插件是在安装过程中需要的(为什么是2015版本,因为2017版本有出错例子)
2.如果你电脑安装了VS2015,且按上面博客教程安装成功,那恭喜你,你不用受折磨了!!!(以前我安装的两次都成功了)
3.如果在安装CUDA一直出现下面错误:安装失败(因为没图,所以用的网上的图,意思一样),那恭喜你,你也中坑了,怎么办?
解决办法:
1.在c盘的Program Files和Program Files(x86)两个文件夹中分别删除NVIDIA GPU Computing Toolkit文件夹。
2.重启电脑
3.重新安装,注意在安装的时候选择自定义安装,且不要勾选Visual Studio Integration(一般安装不成功就是这个原因!!!)
4.安装成功后,后面的过程就按最开始的链接来进行配置路径,在这里注意在博客里,下图圈出的不需要配置
5.在CMD中输入nvcc -V进行验证,出现下面信息代表安装成功,后面的cudnn按教程安装即可,这个不会出错。
特别注意!!!!!下面要说另外一种超简单的安装方法!!!
简单安装!!!
利用anaconda可以直接安装上面所说的所有东西,简直不要太良心,然而我发现的时候,我已经踩坑了,绝望!!!!但是还是分享给大家,具体可以参考下面的博客点我链接.
按这个操作简直不要太良心!!!!!anaconda个人是非常推荐的,具体可以看我的其他博客!!!!!