当前位置:   article > 正文

windows10 显卡GTX1050安装tensorflow-gpu教程以及Tensorflow不同版本要求与CUDA及CUDNN版本对应关系(GPU版本)_gtx1050 tensflowcuda

gtx1050 tensflowcuda

windows10 显卡GTX1050安装tensorflow-gpu教程以及Tensorflow不同版本要求与CUDA及CUDNN版本对应关系(GPU版本)

一、安装准备与对应信息

本人电脑是GTX1050显卡,所选择的版本CUDA9.0+CUDNN7.6+tensorflow-gpu1.10
官网下载太慢,打包在了百度网盘。(链接已更新)
本人分享的网盘:https://pan.baidu.com/s/169tdICW-FJh3l_gYWpaQtw
提取码:d74m
参考官网地址:
Windows端:地址
CPU
在这里插入图片描述
GPU
在这里插入图片描述
还有其他系统的在这个博客可以查看:查看地址

二、安装anaconda

下载地址官网:下载地址
选择相应的系统与版本。
网上与许多安装教程,这里就不给出了。

三、安装cuda9.0

官网各种版本的CUDA下载地址:
下载地址
选择9.0版本
在这里插入图片描述
选择与自己电脑相对应的版本,这里我选择的是本地安装,因为在线安装的话,后面下载可能会不成功。电脑下载实在慢的,还是使用手机下载,我没有科学上网还是很快。
在这里插入图片描述
开始安装
双击安装文件:
在这里插入图片描述
解压地址我选择的默认地址,反正是临时的,没什么关系。
在这里插入图片描述
点击ok开始解压,等待解压完毕。
开始会出现检查系统兼容性,一般都没有问题,点击下一步
(不要在意图片上的版本,按照标题版本来安装)
在这里插入图片描述
重点来了:这里千万不要选默认的精简,这里的精简应该改成全部才对(看下面的小字说明,这就是全家桶),倒不是说安装全家桶不可以,主要是有一个东西的安装会一直导致安装失败。
在这里插入图片描述
特别是这个CUDA下的visual studio integration千万不能选!

原因可能是是电脑里面安装了VS

其余都没什么问题。
在这里插入图片描述
如果最后出现很多未安装和安装失败的字眼,可以将下面的勾选去掉 。
在这里插入图片描述
我就是这样安装成功的!!!
点击下一步,安装位置选择默认的好点,自定义的话可能会出现一些问题导致安装失败。
验证是否安装成功
1.环境变量应该已经自动加好
在这里插入图片描述
2.cmd里查看版本信息nvcc -V
在这里插入图片描述
3.进入到路径下后查看GPU运行时的监测界面
在这里插入图片描述
4.进入下面图片的路径,然后按住‘shift’+鼠标右击,点击’在此处打开powershell窗口‘,输入: .\bandwidthTest.exe 返回Result=PASS
在这里插入图片描述
同理输入:.\deviceQuery.exe 返回:Result=PASS,说明安装成功了。
在这里插入图片描述

四、安装cuDNN7.6

官方各种cuDNN下载地址:下载地址
cuDNN称不上安装,只需要将下载下来的压缩包解压后,将对应文件夹的文件放到CUDA安装路径下的对应文件夹里即可。

我们将其中’bin‘中的’cudnn64_7.dll‘拷贝到cuda9.0安装路径下的bin文件夹中,cuda文件夹的路径如图所示:
在这里插入图片描述
然后将cudnn中include下的’cudnn.h‘拷贝到cuda文件夹中的include的文件夹中
同理,cudnn文件中的’lib/x64’中的‘cudnn.lib’拷贝到cude文件中的’lib/x64’中

五、安装tensorflow-gpu

我们在windows的‘所有程序’中找到安装的Anaconda程序中的Anaconda prompt,单击打开:
在这里插入图片描述
在命令行中使用命令:

conda create -n tensorflow-gpu python=3.6
  • 1

安装python3.6 的tensorflow-gpu环境,可以在anaconda/envs文件夹中看到,和之前创建的python3.6环境在一起。(我之前是重新创建了python3.6)
使用下面命令查看环境是否安装成功:

conda info --envs
  • 1

在这里插入图片描述
出现tensorflow-gpu,发现成功!!

然后输入下面命令来激活tensorflow环境:

conda activate tensorflow-gpu 
  • 1

在这里插入图片描述
前端出现下面字样:
在这里插入图片描述
先更新一下pip工具,输入命令:

python -m pip install --upgrade pip
  • 1

(不是必要的)

然后安装tensorflow-gpu,我们需要指定我们需要的版本,直接安装是最新的不兼容。
安装命令:

pip install tensorflow-gpu==1.10.0
  • 1

如果上面命令不行的话,我指定的豆瓣源:

pip install tensorflow-gpu==1.10.0 -i https://pypi.douban.com/simple
  • 1

下载贼快。
然后安装成功。
关闭tensorflow-gpu环境,输入:

conda deactivate
  • 1

六、在anaconda下载Jupyter-notebook

在anaconda中选择刚刚创建的环境tensorflow-gpu
在这里插入图片描述
点击install(我这个是下载好的,所以显示launch,未下载的应该显示的是install。
多次安装不成功的话,换手机热点试试。)
在这里插入图片描述

安装成功后:
在这里插入图片描述
接下来打开它,使用代码验证gpu是否成功能够运行:
①第一种方法

import tensorflow as tf
tf.test.is_gpu_available()
  • 1
  • 2

②第二种方法

import tensorflow as tf
if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")
  • 1
  • 2
  • 3
  • 4
  • 5

结果展示:
在这里插入图片描述
发现是使用的gpu!!!

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家自动化/article/detail/95636
推荐阅读
相关标签
  

闽ICP备14008679号