win10 TensorFlow-GPU 安装

Win10下Tensorflow(GPU版)

安装Anaconda:

Anaconda指的是一个开源的Python发行版本,其包含了conda、Python等180多个科学包及其依赖项,我理解的Anaconda是跟构建Java项目中的maven包管理工具比较像,我们可以用这个软件进行python的版本管理。

  • 版本选择:Anaconda有众多版本,并不是越新越好,还要考虑兼容性。有些版本的tensorflow支持的是python3.5和3.6,并不支持python3.7,省事起见,还是下载历史版本的anaconda吧。
  • 在这里插入图片描述
    anaconda内置python版本对照表
    根据这张表对比完就可以进行相应版本的下载了
  • 下载地址: 清华镜像或者官网
  • 安装
    下载对应的安装包之后,点击程序一直下一步即可(注意勾选添加到环境变量,不然后面还需要手动添加)
    添加到环境变量
  • 检验是否安装成功
    cmd中输入 conda -V ,显示版本,没有问题 检测是否安装成功
安装TensorFlow-GPU
  1. 创建一个新的虚拟环境,方法是选择 Python 解释器并创建一个 tensorflow 目录来存放它
conda create -n tensorflow pip python=3.6

在这里插入图片描述
等待一会输入 y
在这里插入图片描述
2. 激活虚拟环境

activate tensorflow

在这里插入图片描述
3. 安装tensorflow-GPU
在虚拟环境下执行(要想安装历史版本的可以在后边加上tensorflow-gpu==1.15其中数字对应版本号,很多csdn的教程都是后边没有跟版本号,不加版本号就是默认安装最新版的tesorflow)
稳定版版本的搭配
在这里插入图片描述

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

等待时间较长…(也可以用相关的镜像,提升下载速度参考https://blog.csdn.net/sinat_21591675/article/details/82770360
注意:(安装到一半时会报错,等了半小时,结果报错了,崩溃。。。
从网上查了一下是pip没有升级到最新版本,只能按照以下步骤重新来过

  • 输入指令,进行升级:
    python -m pip install --upgrade pip
  • 升级完成进行安装
    pip install --ignore-installed --upgrade tensorflow-gpu
    在这里插入图片描述
    安装完成可以用以下命令看一下自己安装的版本
pip show tensorflow
安装安装CUDA Toolkit +cuDNN

CUDA是NVIDIA推出的运算平台,CuDNN是专门针对Deep Learning框架设计的一套GPU计算加速方案。在安装之前要查询下最新TensorFLow发行版支持到了哪个版本,同时也要看自己的显卡支持哪个版本的cuda,根据tensorflow的版本和自己的显卡型号是否支持选择相应的cuda和cudnn的安装版本否则后期只能装完成会报错导致不能使用
笔者安装的是tensorflow2.0的版本,笔者的GPU是RTX2060,从英伟达官网可以查到显卡型号的支持,所以我选择了这一套配置
在这里插入图片描述
CUDA安装包下载地址
打开安装包,安装软件会先运行一个系统检查,如果没有软硬件不兼容的情况就能继续进行下一步。若有不兼容情况,系统检查则会报错,无法进行下一步。如果报错了,就最先考虑下是不是自己的显卡不被该版本支持。
在这里插入图片描述
选择自定义,然后不断的下一步安装直到完成安装。
CuDNN下载地址(下载CuDNN需要注册账号,注册过程比较简单)
下载完成,解压压缩包得到三个文件
在这里插入图片描述
将这三个文件复制粘贴到CUDA10.0的安装路径根文件夹下
在这里插入图片描述
添加环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
在这里插入图片描述

安装Visual Studio

在这里插入图片描述
根据这个表格,选择vs2017
安装教程及方法网上的教程很多,可以根据自己的版本需要安装包。如果需要不同安装包的可以评论留下邮箱,我发给你安装包和教程。

验证

在命令行中,进入python,并输入以下代码:
import tensorflow as tf
hello=tf.constant(‘hello,world’)
sess=tf.Session()
print(sess.run(hello))
在这里插入图片描述

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Songgp1024

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值