Tensorflow-gpu 2 安装流程

        重装过很多次Tensorflow了,每次安装前都以为自己已经很熟练了,但是安装过程中还是得搜一下教程,而且还会漏掉一些步骤。所以我决定,将我当前认为较为简单的安装方式写下来,以备不时之需。
1.安装Anaconda
        这一步应该是没有啥注意的,我是官网无脑下载的,且在安装过程中没有设置环境变量。这步直接过了…
2.确定版本
        安装GPU版本需要安装CUDA以及cuDNN,下表给出了2.0及以上版本所对应的Python、cuDNN以及CUDA版本。

Tensorflow-gpu版本Python版本cuDNN版本CUDA版本
2.4.03.6-3.88.011.0
2.3.03.5-3.87.610.1
2.2.03.5-3.87.610.1
2.1.03.5-3.77.610.1
2.0.03.5-3.77.410

        需要注意的是: CUDA的安装只能下载上表中对应的版本。如果想安装Tensorflow2.3.0,却下载了10.2版本的CUDA,则需要重新下载10.1版本。
        以安装Tensorflow2.3.0为例,需要安装10.1版本的CUDA以及7.6版本的cuDNN。
3.安装CUDA
        

进入CUDA的官网会提示安装最新版本的CUDA,想要下载其他版本的CUDA,可以点击下方链接进行下载:

https://developer.nvidia.com/cuda-toolkit-archive

进入网站后找到对应的版本下载安装即可在这里插入图片描述

选择上图所示版本后进入下图界面进行安装即可
在这里插入图片描述
安装过程选择自定义安装后一直下一步即可。下载的文件默认在下图位置:
在这里插入图片描述
安装完成后添加系统变量,在我的电脑右键属性->高级系统设计->环境变量->系统变量 可以看到有如下4行,安装好后就存在的。
在这里插入图片描述
我们在系统变量里找到Path,然后点编辑添加环境变量
在这里插入图片描述
需要将红框中的四项添加进去并移动到最顶端。操作完成后我们Win+R输入cmd,之后输入nvcc -V 可以看到如下提示,则证明安装且配置成功
在这里插入图片描述
4.安装cuDNN
进入下述网站选择对应版本的cuDNN进行安装,安装需要登录账号,登录一下就可以了,也不废话了。如果下载慢可以用镜像源或者直接用迅雷进入下载网址进行下载。

https://developer.nvidia.com/rdp/cudnn-archive

解压好的cuDNN如下图左侧所示,我们将对应的文件夹里的内容复制到cuda10.1文件夹下即可。
在这里插入图片描述
这里给出本次下载所使用的cuDNN 7.6.5版本压缩包,需要的自提:

链接:https://pan.baidu.com/s/1vr-9QBOJbdnxGBulUasqGw
提取码:abcd

5.安装Tensorflow
打开Anaconda Prompt会进入我们的base环境,我们首先使用如下语句新建一个名为tensorflow环境,并且下载Python 3.8:

conda create -n tensorflow Python=3.8

安装成功后我们执行如下代码进入该环境:

activate tensorflow

进入该环境后就可以下载tensorflow-gpu了,我们使用pip以及镜像源进行下载,下载语句如下:

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

等待下载完成即可

6.验证安装
在Anaconda Prompt中的tensorflow环境中输入如下指令安装matplotlib以及notebook:

pip install matplotlib notebook -i https://pypi.douban.com/simple/

安装结束后再输入jupyter notebook 打开网页。之后点击New->Python3新建文文件。
在这里插入图片描述
文件新建后输入如下代码会输出:2.3.0 ,则证明安装成功。

import tensorflow as tf
print(tf.__version__)

在这里插入图片描述
Over!

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值