tensorflow测试GPU可用

因为自己在实现yolo5的过程中其中所需要的的环境是cuda10.1但是在之前安装的是10.0，所以只有自己折腾一下。但是cuda10.1貌似很难安装tensorflow1.X版本，在查阅资料并亲测过之后写下。
二、电脑环境
1.win10专业版
2.CUDA=10.1
3.显卡=1080TI
4.vs2015
三、安装
1.具体的安装步骤就不在这做详细说明，具体可以去看https://blog.csdn.net/weixin_43218120/article/details/107708082
在上面的博客中有详细说明。
2.问题解决
解决安装tensorflow-gpu1.X版本问题只需要在安装tensorflow-gpu之前安装cudatoolkit=10.0的包，具体的安装命令如下：

conda install cudatoolkit=10.0
1
在安装cudatoolkit之后再进行安装tensorflow就成功。
四、测试
1.测试代码

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
print(tf.test.is_gpu_available())print(tf.__version__)

或者

import tensorflow as tf
with tf.device('/gpu:0'):a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')c = tf.matmul(a, b)with tf.Session() as sess:print (sess.run(c))

tensorflow测试GPU可用相关推荐

Ubuntu tensorflow自定义GPU版本op节点
参考:https://blog.csdn.net/qq_27637315/article/details/79114633 windows增加op节点: https://github.com/tens ...
Win10 TensorFlow（gpu）安装详解
Win10 TensorFlow(gpu)安装详解写在前面:TensorFlow是谷歌基于DistBelief进行研发的第二代人工智能学习系统,其命名来源于本身的运行原理.Tensor(张量)意味着 ...
深度学习笔记：windows+tensorflow 指定GPU占用内存（解决gpu爆炸问题）
目录文章目录目录前言一.指定显卡二.限制GPU用量 1.设置使用GPU的百分比进行配置,使用30%的GPU 设置session 2.GPU按需使用三.指定GPU并且限制GPU用量指定第 ...
使用TensorFlow，GPU和Docker容器进行深度学习
摘要: 数据科学家使用GPU来提高TensorFlow的计算速度,但GPU价格昂贵,也需要对其所占用的资源进行认真的管理.本文将带你来一起解决这一问题. 在过去的几个月中,我和多个企业的数据科学团队进 ...
tensorflow with gpu 环境配置
1.准备工作 1.1 确保GPU驱动已经安装 lspci | grep -i nvidia 通过此命令可以查看GPU信息,测试机已经安装GPU驱动 nvidia-smi 可以查看英伟达显卡信息 1.2 ...
gpu跑普通python程序_普通电脑PC怎样跑TensorFlow的GPU模式
展开全部首先需要看你的PC配置是否32313133353236313431303231363533e4b893e5b19e31333363373736够,TF的GPU模式只支持N卡,然后计算能力高于 ...
Windows10下Tensorflow启用GPU加速，显卡GTX1060，踩坑记录
因为需要用到tensorflow学习深度学习,所以有N卡就想开启GPU加速,结果各种坑 1.安装VS和Python环境 (不用VS的可以不安装,使用其他工具也是一样的) 这里使用VS2019作为开发 ...
安装tensorflow的GPU版本（详细图文教程）--CUDA11.6的安装
文章目录 TensorFlow简介 TensorFlow是什么 tensorflow版本变迁 tensorflow 2.0 架构安装过程常用IDE安装 python3.9的安装 Anaconda的 ...
tensorflow的GPU加速计算
参考 tensorflow的GPU加速计算 - 云+社区 - 腾讯云一.概述 tensorflow程序可以通过tf.device函数来指定运行每一个操作的设备,这个设备可以是本地的CPU或者G ...

tensorflow测试GPU可用

tensorflow测试GPU可用相关推荐

最新文章

热门文章