欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

在VS2017的python模块中安装tensorflow1.14.0_gpu

程序员文章站 2022-07-13 21:57:44
...

在VS2017的python模块中安装tensorflow1.14.0_gpu

有针对性的文章

本人深度学习小白一个,第一次安装python环境的tensorflow(之前装的是C#版本的),本着有需求的想法就硬着头皮搞起了gpu版本。上面我也说了,这是一篇有针对性的文章,毕竟每个人的电脑环境都不一样,我写这篇博客也单纯为了记录一下而已。第一篇博客,如有说的不清楚的,写的不清楚的还请见谅。

我的安装环境

  1. VS2017(版本15.9)
  2. python3.6_64bit(直接在VS里装的)
  3. CUDA v10.0(cudart64_100.dll 10.0.130)
  4. cudnn v7.6.5.32zip(cudnn64_7.dll 10.0.145)
  5. tensorflow1.14.0_gpu
  6. 显卡:GTX1650(移动版)

步骤

1、安装VS2017;
2、在Visual Studio Installer中安装python(电脑之前未安装python),默认勾选安装就行了,我默认安装后的版本就是python3.6_64bit;
在VS2017的python模块中安装tensorflow1.14.0_gpu
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/. 应该是阿里云的镜像库,速度快到爆炸,忘了是在哪里找到的了,在此也感谢分享此网站的人和这个网站。进入网站直接在当前页面下拉找到对应版本点击下载即可。
在VS2017的python模块中安装tensorflow1.14.0_gpu
下载完成之后将文件放在指定路径下,我的放在这里:C:\Program Files (x86)\Microsoft Visual Studio\Shared\Python36_64,然后打开visual studio,新建一个python项目,在右侧的项目资源管理器中右击python环境,点击查看所有python环境:
在VS2017的python模块中安装tensorflow1.14.0_gpu
在python3.6高亮下选择下拉框中Packages(PyPI),在搜索框中输入刚才下载好的tensorflow文件名(带后缀),然后点击红框中命令即安装命令进行安装,默认会安装所有依赖库,但不知道什么原因,我再安装的时候有一个依赖库没装就是numpy,如果遇到这种情况就去刚才的镜像库找符合的版本下载安装即可,安装方法同上。
在VS2017的python模块中安装tensorflow1.14.0_gpu
显示安装成功就行了。

测试

在网上随便找了一段测试代码进行测试:

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)

输出结果如下图,看到结果后留下了不争气的泪水,至此环境配置完毕。
在VS2017的python模块中安装tensorflow1.14.0_gpu
GPU版的tensorflow在安装CUDA和cudnn时版本是有差别的,网上都能搜到相匹配的版本;其实各种环境之间都有所依赖和匹配,这里说的不清楚和不对的地方还请大佬们见谅和指正。
CPU版本的tensorflow安装比GPU版的简单,网上也很多教程,希望这篇文章对你们有帮助,哪怕是一丁点帮助。
欢迎各路大佬回复讨论,好让我这个小白多学习学习。