tensorflow-GPU安装文档

tensorflow-GPU安装文档

1.Anaconda安装

官网下载地址

https://www.anaconda.com/products/individual

安装流程

在这里插入图片描述

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-E9YQxiW9-1646054229446)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228205234056.png)]

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-ba3AAwqj-1646054229447)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228205251563.png)]

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-haCfDu05-1646054229447)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228205312256.png)]

注意:此处要****勾选第一项Add Anaconda3 to my PATH environment variable****

2.安装tensorflow

1、打开Anaconda Prompt

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-8ToEOvGJ-1646054229448)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228205501192.png)]

2.在Anaconda Prompt中更换清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

即可添加 Anaconda Python 免费仓库。

3.新建一个TF2.1的环境 用python3.7的版本

conda create -n TF2.1 python=3.7

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-k7DUvNQH-1646054229448)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228210141991.png)]

4.安装tensorflow

首先进入tensorflow环境

conda activate TF2.1

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-XTv6KhzU-1646054229449)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228210245109.png)]

5.安装英伟达的SDK10.1版本

conda install cudatoolkit=10.1

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Qb70zGAU-1646054229449)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228210313188.png)]

出现以上界面一路y即可

同理,安装英伟达深度学习包软件

conda install cudnn=7.6

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-nVHDXhfe-1646054229450)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228210332723.png)]

如果出错,说明电脑不支持,或者有可能是源没挂对,自行百度解决

3.安装tensorflow

pip install tensorflow==2.1

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-tJwiGVAY-1646054229450)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228210349403.png)]

  1. 进入 python 验证是否安装成功

    进入python

    输入import tensorflow as tf 导入

    然后输入 tf.version 查看版本

(TF2.1) C:\Users\tsl115>python
Python 3.7.11 (default, Jul 27 2021, 09:42:29) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
2022-02-28 20:39:57.511027: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
>>> tf__version
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'tf__version' is not defined
>>> tf.__version__
'2.1.0'
>>>

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-MsmMLoS8-1646054229450)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228210635304.png)]

如果显示和我差不多,说明安装成功

3.安装pycharm社区版

直接进官网下载社区版即可,具体百度就有,不会csdn上有很多教程

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-ZGeD39sK-1646054229451)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228210750659.png)]

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-WfspIp6o-1646054229451)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228210817939.png)]

需要注意这一步,其他傻瓜式

  1. 进入pycharm进行配置测试

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-PojzV95R-1646054229451)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228210923090.png)]

在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

输入代码进行测试

import tensorflow as tf

tensorflow_version = tf.__version__
gpu_available = tf.test.is_gpu_available()

print("tensorflow version:", tensorflow_version, "\tGPU available:", gpu_available)

a = tf.constant([1.0, 2.0], name="a")
b = tf.constant([1.0, 2.0], name="b")
result = tf.add(a, b, name="add")
print(result)

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-fA5GIZul-1646054229453)(tensorflow-GPU%E5%AE%89%E8%A3%85%E6%96%87%E6%A1%A3.assets/image-20220228211203290.png)]

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值