WIN10 环境下 Tensorflow 检测不到GPU (GTX1050笔记本版可以安装CUDA)

本文主要是介绍WIN10 环境下 Tensorflow 检测不到GPU (GTX1050笔记本版可以安装CUDA),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

在python环境下执行下列代码

tf.config.experimental.list_physical_devices(device_type='GPU')

返回的结果是一个空列表,说明检测不到GPU。

去Tensorflow官网,发现显卡需要支持CUDA才行。
在这里插入图片描述

网上查找资料后发现,原因是我的笔记本电脑的显卡是GTX 1050,不支持CUDA
在这里插入图片描述(截图没有截全)
(官网查询:https://developer.nvidia.com/cuda-gpus)

笔记本版的GTX1050不支持CUDA,但是桌面版的支持。

所以,我的笔记本上的Tensorflow用不了GPU。。。

剧情出现反转,

虽然官网上的CUDA支持列表上没有笔记本版的GTX1050。但是通过英伟达的控制面板里可以看到有CUDA核心数,
在这里插入图片描述并且网上也有人说能用,我就装了一下试试。结果成功了。
我用tensorflow可以调用GPU进行运算了。
输入:

tf.config.experimental.list_physical_devices(device_type='GPU')

返回:
在这里插入图片描述
这里说一下注意事项吧,
先去Tensor flow的官网看一下GPU支持的要求:
(这里只说Tensor flow2.0之后的版本,2.0之前的没用过不太清楚)
在这里插入图片描述首先你的显卡支持CUDA(鉴于英伟达官网有点坑,可以去英伟达控制面板查看是否支持CUDA,不过这个方法不知道可靠性如何)
也就是Tensorflow要用2.1.0及以上版本,CUDA要用10.1版本,显卡驱动要418.x及以上版本,并且要安装对应的cuDNN。

说一下我的配置:Win10系统下,Tensoflow2.3.0 +CUDA10.1+cuDNN+visual studio 2019
(cuDNN要选择与CUDA对应的版本。visual studio要提前安装,应该是用来编译cuda程序的)
(我之前用的Tensorflow2.0.0版本,但是检测不到GPU,后来更新到2.3.0就好了。)
具体的安装操作,网上有很多帖子,不做赘述。。。

最后说一下,为什么要用GPU,
在Tensorflow的环境下,用CPU和GPU运行两段相同张量运算程序,耗时如下:
在这里插入图片描述
在这里插入图片描述

加大运算量:
在这里插入图片描述
这运行时间差了不是一两倍的问题啊!

这篇关于WIN10 环境下 Tensorflow 检测不到GPU (GTX1050笔记本版可以安装CUDA)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/356967

相关文章

电脑找不到mfc90u.dll文件怎么办? 系统报错mfc90u.dll丢失修复的5种方案

《电脑找不到mfc90u.dll文件怎么办?系统报错mfc90u.dll丢失修复的5种方案》在我们日常使用电脑的过程中,可能会遇到一些软件或系统错误,其中之一就是mfc90u.dll丢失,那么,mf... 在大部分情况下出现我们运行或安装软件,游戏出现提示丢失某些DLL文件或OCX文件的原因可能是原始安装包

使用Python实现IP地址和端口状态检测与监控

《使用Python实现IP地址和端口状态检测与监控》在网络运维和服务器管理中,IP地址和端口的可用性监控是保障业务连续性的基础需求,本文将带你用Python从零打造一个高可用IP监控系统,感兴趣的小伙... 目录概述:为什么需要IP监控系统使用步骤说明1. 环境准备2. 系统部署3. 核心功能配置系统效果展

Win安装MySQL8全过程

《Win安装MySQL8全过程》:本文主要介绍Win安装MySQL8全过程,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录Win安装mysql81、下载MySQL2、解压文件3、新建文件夹data,用于保存数据库数据文件4、在mysql根目录下新建文件my.ini

判断PyTorch是GPU版还是CPU版的方法小结

《判断PyTorch是GPU版还是CPU版的方法小结》PyTorch作为当前最流行的深度学习框架之一,支持在CPU和GPU(NVIDIACUDA)上运行,所以对于深度学习开发者来说,正确识别PyTor... 目录前言为什么需要区分GPU和CPU版本?性能差异硬件要求如何检查PyTorch版本?方法1:使用命

IntelliJ IDEA 中配置 Spring MVC 环境的详细步骤及问题解决

《IntelliJIDEA中配置SpringMVC环境的详细步骤及问题解决》:本文主要介绍IntelliJIDEA中配置SpringMVC环境的详细步骤及问题解决,本文分步骤结合实例给大... 目录步骤 1:创建 Maven Web 项目步骤 2:添加 Spring MVC 依赖1、保存后执行2、将新的依赖

解决Maven项目idea找不到本地仓库jar包问题以及使用mvn install:install-file

《解决Maven项目idea找不到本地仓库jar包问题以及使用mvninstall:install-file》:本文主要介绍解决Maven项目idea找不到本地仓库jar包问题以及使用mvnin... 目录Maven项目idea找不到本地仓库jar包以及使用mvn install:install-file基

最详细安装 PostgreSQL方法及常见问题解决

《最详细安装PostgreSQL方法及常见问题解决》:本文主要介绍最详细安装PostgreSQL方法及常见问题解决,介绍了在Windows系统上安装PostgreSQL及Linux系统上安装Po... 目录一、在 Windows 系统上安装 PostgreSQL1. 下载 PostgreSQL 安装包2.

Maven如何手动安装依赖到本地仓库

《Maven如何手动安装依赖到本地仓库》:本文主要介绍Maven如何手动安装依赖到本地仓库问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录一、下载依赖二、安装 JAR 文件到本地仓库三、验证安装四、在项目中使用该依赖1、注意事项2、额外提示总结一、下载依赖登

如何在Mac上安装并配置JDK环境变量详细步骤

《如何在Mac上安装并配置JDK环境变量详细步骤》:本文主要介绍如何在Mac上安装并配置JDK环境变量详细步骤,包括下载JDK、安装JDK、配置环境变量、验证JDK配置以及可选地设置PowerSh... 目录步骤 1:下载JDK步骤 2:安装JDK步骤 3:配置环境变量1. 编辑~/.zshrc(对于zsh

Python如何自动生成环境依赖包requirements

《Python如何自动生成环境依赖包requirements》:本文主要介绍Python如何自动生成环境依赖包requirements问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑... 目录生成当前 python 环境 安装的所有依赖包1、命令2、常见问题只生成当前 项目 的所有依赖包1、