在VS2017的python模块中安装tensorflow1.14.0_gpu
有针对性的文章
本人深度学习小白一个,第一次安装python环境的tensorflow(之前装的是C#版本的),本着有需求的想法就硬着头皮搞起了gpu版本。上面我也说了,这是一篇有针对性的文章,毕竟每个人的电脑环境都不一样,我写这篇博客也单纯为了记录一下而已。第一篇博客,如有说的不清楚的,写的不清楚的还请见谅。
我的安装环境
- VS2017(版本15.9)
- python3.6_64bit(直接在VS里装的)
- CUDA v10.0(cudart64_100.dll 10.0.130)
- cudnn v7.6.5.32zip(cudnn64_7.dll 10.0.145)
- tensorflow1.14.0_gpu
- 显卡:GTX1650(移动版)
步骤
1、安装VS2017;
2、在Visual Studio Installer中安装python(电脑之前未安装python),默认勾选安装就行了,我默认安装后的版本就是python3.6_64bit;
3、安装CUDA和cudnn;安装步骤和版本完全参考了下面这篇博文,在此感谢这个博主,讲得很详细,这里我就不复述了(只需要看安装CUDA和cudnn部分就行了),需要说明的一点是配置环境变量,我安装好CUDA之后系统默认添加了环境变量,如果你的电脑没有默认添加就按照下面这篇手动添加即可:
https://blog.csdn.net/wumenglu1018/article/details/96473096.
4、下载安装tensorflow1.14.0_gpu,方法很多,我是在网上找的whl文件,附上下载链接:http://mirrors.aliyun.com/pypi/simple/tensorflow-gpu/. 应该是阿里云的镜像库,速度快到爆炸,忘了是在哪里找到的了,在此也感谢分享此网站的人和这个网站。进入网站直接在当前页面下拉找到对应版本点击下载即可。
下载完成之后将文件放在指定路径下,我的放在这里:C:\Program Files (x86)\Microsoft Visual Studio\Shared\Python36_64,然后打开visual studio,新建一个python项目,在右侧的项目资源管理器中右击python环境,点击查看所有python环境:
在python3.6高亮下选择下拉框中Packages(PyPI),在搜索框中输入刚才下载好的tensorflow文件名(带后缀),然后点击红框中命令即安装命令进行安装,默认会安装所有依赖库,但不知道什么原因,我再安装的时候有一个依赖库没装就是numpy,如果遇到这种情况就去刚才的镜像库找符合的版本下载安装即可,安装方法同上。
显示安装成功就行了。
测试
在网上随便找了一段测试代码进行测试:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
res1 = sess.run(hello)
print(res1)
a = tf.constant(10)
b = tf.constant(32)
res2 = sess.run(a + b)
print(res2)
输出结果如下图,看到结果后留下了不争气的泪水,至此环境配置完毕。
GPU版的tensorflow在安装CUDA和cudnn时版本是有差别的,网上都能搜到相匹配的版本;其实各种环境之间都有所依赖和匹配,这里说的不清楚和不对的地方还请大佬们见谅和指正。
CPU版本的tensorflow安装比GPU版的简单,网上也很多教程,希望这篇文章对你们有帮助,哪怕是一丁点帮助。
欢迎各路大佬回复讨论,好让我这个小白多学习学习。
推荐阅读
-
在python中安装basemap的教程
-
Linux系统在CPAN中安装Perl模块的方法
-
Python中time模块与datetime模块在使用中的不同之处
-
Python3.6中Twisted模块安装的问题与解决
-
在Python中关于使用os模块遍历目录的实现方法
-
python在ubuntu中的几种安装方法(小结)
-
Python中heapq堆的一些基本操作!!——————Python内置模块heapq(无需安装)
-
安装完dlib以及face_recognition,但是在python IDE中无法导入的问题
-
Python环境下PyQt5的安装以及QT Desinger和PyUIC在Pycharm中的配置
-
在VS2017的python模块中安装tensorflow1.14.0_gpu