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

编程知识 更新时间:2023-04-17 02:26:14

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

    • 一、安装准备与对应信息
    • 二、安装anaconda
    • 三、安装cuda9.0
    • 四、安装cuDNN7.6
    • 五、安装tensorflow-gpu
    • 六、在anaconda下载Jupyter-notebook

一、安装准备与对应信息

本人电脑是GTX1050显卡,所选择的版本CUDA9.0+CUDNN7.6+tensorflow-gpu1.10
官网下载太慢,打包在了百度网盘。(链接已更新)
本人分享的网盘:https://pan.baidu/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

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

conda info --envs


出现tensorflow-gpu,发现成功!!

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

conda activate tensorflow-gpu 


前端出现下面字样:

先更新一下pip工具,输入命令:

python -m pip install --upgrade pip

(不是必要的)

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

pip install tensorflow-gpu==1.10.0

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

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

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

conda deactivate

六、在anaconda下载Jupyter-notebook

在anaconda中选择刚刚创建的环境tensorflow-gpu

点击install(我这个是下载好的,所以显示launch,未下载的应该显示的是install。
多次安装不成功的话,换手机热点试试。)

安装成功后:

接下来打开它,使用代码验证gpu是否成功能够运行:
①第一种方法

import tensorflow as tf
tf.test.is_gpu_available()

②第二种方法

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")

结果展示:

发现是使用的gpu!!!

更多推荐

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

本文发布于:2023-04-13 17:40:00,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/8c131084c700300075e1ab29629643d0.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:版本   显卡   关系   教程   tensorflow

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!

  • 73420文章数
  • 14阅读数
  • 0评论数