Tensorflow-gpu版本安装

安装分为3个步骤

1.安装CUDA

首先要查看自己电脑的显卡支持的CUDA版本
打开英伟达控制面板打开帮助点击系统设置,查看自己显卡支持的最高版本的CUDA
在这里插入图片描述
该版本以下的CUDA都可以安装
下载连接https://developer.nvidia.com/cuda-toolkit-archive
如何选择版本?
可以到Tensorflow官网看相关的Python对应Tensorflow版本,以及cuDNN版本
https://tensorflow.google.cn/install/source_windows#gpu
CUDA所占空间比较大,但是也不建议在安装是使用推荐安装,点击下面的自定义安装,将所有组件都勾选,这样可以防止在最后运行时出现某些dll文件的缺失,带来不必要的麻烦🐱‍🐉

2. 安装cuDNN

对应上面的CUDA的版本来进行安装相应的版本
https://developer.nvidia.com/rdp/cudnn-archive
安装解压后将文件复制到CUDA安装的目录下,器目录下有对应的include,lib文件夹,直接复制到CUDA的文件夹下,他会自动将各相同文件夹下面的文件添加到一起的

3. 安装tensorflow-gpu

选对Python版本后,就可以安装相应的tensorflow版本去pip了
例如:
安装1.7.0版本的tensorflow需要CUDA9.0和cuDNN7.0,Python版本我选择3.6。这样就可以运行了。

import tensorflow as tf
import os

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'# 这里是CPU版本中的某个控件不使用,如果下载的是GPU版本,加上该句就不会再控制台输出错误信息了

hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello).decode())

输出Hello, TensorFlow!就是配置成功了

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值