win10(家庭版)+tensorflow-gpu环境搭建(小白篇)

想写这篇文章主要还是想自己总结的同时,可以方便一下别人不踩或者少踩坑,初学者,有错一定请指正我。

工具准备

1.win10系统
2.cuda+cudnn
3.anaconda+pycharm(我是用pycharm的)
系统就不多说了,下面说说cuda+cudnn

cuda+cudnn下载安装

cuda和cudnn在nvidia官网就可以下载,最需要注意的是版本的选择,因为两者版本需要匹配,他们两者跟python环境也需要匹配,跟tensorflow-gpu也需要匹配。
https://developer.nvidia.comcuda及cudnn下载链接
系统支持的cuda版本可在nvidia控制面板-----系统信息----Nvcuda.dll看到
在这里插入图片描述
这里可以看我的系统支持的是cuda10.1及以下版本cuda,查看了之后可以在官网找到相对应的cuda版本及cudnn版本,记住还要看看python版本,具体的步骤我就不多说了,文末有我装环境是参考的博主的博客链接,有说明。
正确安装之后在cmd输入nvcc -V 若出现下图,就说明正常了
在这里插入图片描述

工具准备好之后就可以开始安装了,注意一定要版本匹配!版本匹配!版本匹配!cuda和cudnn以及其python环境都要匹配。

安装

安装之前保证网络状态良好。
如果你cuda,cudnn匹配了安装就非常简单,我是选择借助anaconda安装,创建一个虚拟环境,可以支持多个框架(小白神器),还方便删除而不影响其他。
用管理员身份打开anaconda的prompt,因为我不清楚你电脑情况,所以用管理员身份就对了,不然会白给。

在这里插入图片描述
然后我以我的环境做例子,
在prompt中输入

conda create -u tensorflow python=3.6

然后通过
activate tensorflow

激活环境
然后就可以输入

pip install --ignore-installed --upgrade tensorflow-gpu

过程可能有waining或者error,需要更新一pip,跟着提示走就可以,但是出现这个

在这里插入图片描述
千万不要更新,不过不同电脑情况可能不一样,我当初更新之后anaconda直接用不了,conda也用不了,python环境被破坏,跟
https://github.com/conda/conda/issues/8172conda命令无法使用
情况类似。只能把所有东西删除重新安装anaconda。
装完就可以在prompt进入python然后–import tensorflow如果没有报错就可以开始玩耍了

pycharm调用

在pycharm中点project------设置-----如图,选择anaconda中env在这里插入图片描述

总结

有了前车之鉴,后来学paddle的时候就很快搞定环境问题。
另外要保持网络良好,以及下载慢可以换国内源,操作我就不加了,很快就可以查到。
最近装paddle好像conda命令可搜索快速源(不太清楚)。
还有可能需要c++运行库啥的,有个vs就特快。
因为是之前在学校装的环境,所以有些错误及解决过程没记录就不能摆出来(我忘记了),有点sorry。
最后贴上我曾经参考的两篇文章,因为当时花了将近一个星期,查看了很多文章才搞定,
然而只收藏了两,尴尬。
参考:https://blog.csdn.net/weixin_40802676/article/details/94753951
https://blog.csdn.net/qq_37960402/article/details/89178208

发布了4 篇原创文章 · 获赞 0 · 访问量 116

猜你喜欢

转载自blog.csdn.net/qq_43743037/article/details/104242758