TensorFlow测试CPU、GPU

最新推荐文章于 2025-06-30 10:04:09 发布

￡沐月

最新推荐文章于 2025-06-30 10:04:09 发布

阅读量2.6k

点赞数 3

CC 4.0 BY-SA版权

文章标签： tensorflow 深度学习 python

本文链接：https://blog.csdn.net/weixin_43990532/article/details/124581628

1.查看当前Tensorflow版本

import tensorflow as tf
print(tf.__version__)

会得到自己的tensorflow的版本号
在这里插入图片描述

2.查看当前主机运行的设备

import tensorflow as tf

gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
cpus = tf.config.experimental.list_physical_devices(device_type='CPU')
print(gpus)
print(cpus)

在这里插入图片描述

3.查看GPU是否可用，指定在CPU/GPU运行

import tensorflow as tf

# 指定在cpu上运行
with tf.device

最低0.47元/天解锁文章

200万优质内容无限畅学

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

￡沐月

关注关注

3
点赞
踩
8

收藏

觉得还不错? 一键收藏
0
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
举报

举报

TensorFlow_CPU版安装测试（Anaconda，Jupyter）

JING博客-风起微雨落

12-25

533

conda方式的环境解析很成功：numpy版本1.26.4，不存在兼容性问题。训练一遍大概需要20分钟！

cuda10.2安装GPU的Tensorflow

年少的勇气已经用完，剩下的就是三思而后行

04-26

2514

# 安装 pip install tensorflow-gpu==2.5.0 测试： # 查看tf版本 import tensorflow as tf print(tf.__version__) # 查看主机上运行的设备 import tensorflow as tf gpus = tf.config.experimental.list_physical_devices(device_type='GPU') cpus = tf.config.experimental.list_physical_devi

参与评论您还未登录，请先登录后发表或查看评论

TensorFlow 2.1.0 CPU版本实践指南

weixin_42124497的博客

06-09

788

在本章中，我们将探索TensorFlow 2.1.0 CPU版本的核心特性。TensorFlow是由Google开发的一个开源机器学习框架，它被广泛应用于深度学习领域。2.1.0版本的发布，标志着该框架在易用性、性能优化及社区支持方面的一系列提升。首先，我们来讨论TensorFlow 2.1.0 CPU版本对新手用户友好的特性，比如Eager Execution模式的引入，它使得动态图的开发变得更加直观和容易调试。

TensorFlow实战：从数据读取到模型训练与测试

最新发布

weixin_42596214的博客

06-30

876

在深度学习项目中，数据读取是影响模型性能和训练效率的重要环节。TensorFlow通过其内置的数据读取和处理机制，提供了一种高效的方式来准备和输入数据。了解和掌握数据读取机制，对于任何希望深入TensorFlow的开发者来说，都是基础且必备的知识。在机器学习和深度学习项目中，有效地从原始数据中提取和处理信息是构建模型的前提。随着数据集规模的增长和复杂度的提升，如何高效地加载和处理数据成为了瓶颈问题。为了解决这一问题，TensorFlow推出了tf.data。

tensorflow1.12支持cuda10

01-11

自编译tensorflow： 1.python3.5,tensorflow1.12; 2.支持cuda10.0,cudnn7.3.1,TensorRT-5.0.2.6-cuda10.0-cudnn7.3; 3.无mkl支持; 软硬件硬件环境：Ubuntu16.04,GeForce GTX 1080 TI 配置信息： hp@dla:~/work/ts_compile/tensorflow$ ./configure WARNING: --batch mode is deprecated. Please instead explicitly shut down your Bazel server using the command "bazel shutdown". You have bazel 0.19.1 installed. Please specify the location of python. [Default is /usr/bin/python]: /usr/bin/python3 Found possible Python library paths: /usr/local/lib/python3.5/dist-packages /usr/lib/python3/dist-packages Please input the desired Python library path to use. Default is [/usr/local/lib/python3.5/dist-packages] Do you wish to build TensorFlow with XLA JIT support? [Y/n]: XLA JIT support will be enabled for TensorFlow. Do you wish to build TensorFlow with OpenCL SYCL support? [y/N]: No OpenCL SYCL support will be enabled for TensorFlow. Do you wish to build TensorFlow with ROCm support? [y/N]: No ROCm support will be enabled for TensorFlow. Do you wish to build TensorFlow with CUDA support? [y/N]: y CUDA support will be enabled for TensorFlow. Please specify the CUDA SDK version you want to use. [Leave empty to default to CUDA 10.0]: Please specify the location where CUDA 10.0 toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]: /usr/local/cuda-10.0 Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 7]: 7.3.1 Please specify the location where cuDNN 7 library is installed. Refer to README.md for more details. [Default is /usr/local/cuda-10.0]: Do you wish to build TensorFlow with TensorRT support? [y/N]: y TensorRT support will be enabled for TensorFlow. Please specify the location where TensorRT is installed. [Default is /usr/lib/x86_64-linux-gnu]://home/hp/bin/TensorRT-5.0.2.6-cuda10.0-cudnn7.3/targets/x86_64-linux-gnu Please specify the locally installed NCCL version you want to use. [Default is to use https://github.com/nvidia/nccl]: Please specify a list of comma-separated Cuda compute capabilities you want to build with. You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus. Please note that each additional compute capability significantly increases your build time and binary size. [Default is: 6.1,6.1,6.1]: Do you want to use clang as CUDA compiler? [y/N]: nvcc will be used as CUDA compiler. Please specify which gcc should be used by nvcc as the host compiler. [Default is /usr/bin/gcc]: Do you wish to build TensorFlow with MPI support? [y/N]: No MPI support will be enabled for TensorFlow. Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified [Default is -march=native -Wno-sign-compare]: Would you like to interactively configure ./WORKSPACE for Android builds? [y/N]: Not configuring the WORKSPACE for Android builds. Preconfigured Bazel build configs. You can use any of the below by adding "--config=" to your build command. See .bazelrc for more details. --config=mkl # Build with MKL support. --config=monolithic # Config for mostly static monolithic build. --config=gdr # Build with GDR support. --config=verbs # Build with libverbs support. --config=ngraph # Build with Intel nGraph support. --config=dynamic_kernels # (Experimental) Build kernels into separate shared objects. Preconfigured Bazel build configs to DISABLE default on features: --config=noaws # Disable AWS S3 filesystem support. --config=nogcp # Disable GCP support. --config=nohdfs # Disable HDFS support. --config=noignite # Disable Apacha Ignite support. --config=nokafka # Disable Apache Kafka support. --config=nonccl # Disable NVIDIA NCCL support. Configuration finished 编译： bazel build --config=opt --verbose_failures //tensorflow/tools/pip_package:build_pip_package 卸载已有tensorflow： hp@dla:~/temp$ sudo pip3 uninstall tensorflow 安装自己编译的成果： hp@dla:~/temp$ sudo pip3 install tensorflow-1.12.0-cp35-cp35m-linux_x86_64.whl

win10下cuda10.2配置tensorflow2.x

mczhao0502的博客

01-08

1803

目前，tensorflow官方还未发布cuda10.2对应的tensorflow版本，通过以下方法可以解决改问题亲测有效的配置方法： windows下需要把cuda安装目录下的cudart64_102.dll复制一份重命名为cudart64_101.dll，具体目录是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin\cudart64_101.dll， ...

Ubuntu tensorflow2.X 调用cuda10.2

unsterbliche的博客

04-11

1014

已经安装了CUDA 10.2，然后才发现TensorFlow 2.2只支持CUDA 10.1，CUDA10.2没有对应的tensorflow版本。采取下面的办法建一个软链接即可。注意，我的cuda安装在conda虚拟环境里，不使用根目录里安装的cuda ln -s ~/anaconda3/envs/pytorch/lib/libcudart.so.10.2.89 ~/anaconda3/envs/pytorch/lib/libcudart.so.10.1 相关：conda创建虚拟环境和

CUDA10.2下安装tensorflow-gpu2.0.0版本注意事项

热门推荐

yuanzhen_0128的博客

05-31

1万+

CUDA10.2下安装tensorflow-gpu2.0.0版本注意事项最近新入手了一台i5-10200U的CPU，有一个极为鸡肋的MX250显卡（2G），且经得起造的ThinkPad T490笔记本电脑，为将来的移动办公做准备。今天突然心血来潮想试试它能不能用cuda的加速，去了NVIDIA的官网查询后，发现这块鸡肋N卡还真的支持cuda加速，就这样花了大半天的时间将cuda安装好后，并进行了一个CNN的网络训练。一、安装准备需要在你的NVIDIA控制面板→右下角系统信息→组件，去查看你的NVCUD

tensorflow2.5.0-GPU测试工程

07-27

**TensorFlow 2.5.0 GPU版本的测试与应用** 在深度学习领域，TensorFlow是一个广泛使用的开源库，它提供了强大的工具来构建和训练神经网络模型。TensorFlow 2.5.0版本进一步优化了性能，特别是对于配备GPU的系统，...

tensorflow-cpu gpu安装教程1

08-03

在安装TensorFlow时，根据硬件配置可以选择CPU版本或GPU版本。本教程将详细讲解在Ubuntu 16.04系统上安装TensorFlow CPU版的两种方法。首先，确保系统已安装GCC编译器，推荐使用版本为4.4或更高。GCC是编译Python...

tensorflow查看GPU，CPU，指定设备运行

猛犸象

10-31

708

tensorflow查看GPU，CPU，指定设备运行

TensorFlow-gpu测试

12-17

TensorFlow-gpu版本测试代码，用于测试当前设备所使用的GPU类型

tensorflow-1.12支持cuda10.0

01-11

自编译tensorflow： 1.python3.5,tensorflow1.12; 2.支持cuda10.0,cudnn7.3.1,TensorRT-5.0.2.6-cuda10.0-cudnn7.3; 3.支持mkl,无MPI; 软硬件硬件环境：Ubuntu16.04,GeForce GTX 1080 配置信息： hp@dla:~/work/ts_compile/tensorflow$ ./configure WARNING: --batch mode is deprecated. Please instead explicitly shut down your Bazel server using the command "bazel shutdown". You have bazel 0.19.1 installed. Please specify the location of python. [Default is /usr/bin/python]: /usr/bin/python3 Found possible Python library paths: /usr/local/lib/python3.5/dist-packages /usr/lib/python3/dist-packages Please input the desired Python library path to use. Default is [/usr/local/lib/python3.5/dist-packages] Do you wish to build TensorFlow with XLA JIT support? [Y/n]: XLA JIT support will be enabled for TensorFlow. Do you wish to build TensorFlow with OpenCL SYCL support? [y/N]: No OpenCL SYCL support will be enabled for TensorFlow. Do you wish to build TensorFlow with ROCm support? [y/N]: No ROCm support will be enabled for TensorFlow. Do you wish to build TensorFlow with CUDA support? [y/N]: y CUDA support will be enabled for TensorFlow. Please specify the CUDA SDK version you want to use. [Leave empty to default to CUDA 10.0]: Please specify the location where CUDA 10.0 toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]: /usr/local/cuda-10.0 Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 7]: 7.3.1 Please specify the location where cuDNN 7 library is installed. Refer to README.md for more details. [Default is /usr/local/cuda-10.0]: Do you wish to build TensorFlow with TensorRT support? [y/N]: y TensorRT support will be enabled for TensorFlow. Please specify the location where TensorRT is installed. [Default is /usr/lib/x86_64-linux-gnu]:/home/hp/bin/TensorRT-5.0.2.6-cuda10.0-cudnn7.3/targets/x86_64-linux-gnu Please specify the locally installed NCCL version you want to use. [Default is to use https://github.com/nvidia/nccl]: Please specify a list of comma-separated Cuda compute capabilities you want to build with. You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus. Please note that each additional compute capability significantly increases your build time and binary size. [Default is: 6.1,6.1,6.1]: Do you want to use clang as CUDA compiler? [y/N]: nvcc will be used as CUDA compiler. Please specify which gcc should be used by nvcc as the host compiler. [Default is /usr/bin/gcc]: Do you wish to build TensorFlow with MPI support? [y/N]: No MPI support will be enabled for TensorFlow. Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified [Default is -march=native -Wno-sign-compare]: Would you like to interactively configure ./WORKSPACE for Android builds? [y/N]: Not configuring the WORKSPACE for Android builds. Preconfigured Bazel build configs. You can use any of the below by adding "--config=" to your build command. See .bazelrc for more details. --config=mkl # Build with MKL support. --config=monolithic # Config for mostly static monolithic build. --config=gdr # Build with GDR support. --config=verbs # Build with libverbs support. --config=ngraph # Build with Intel nGraph support. --config=dynamic_kernels # (Experimental) Build kernels into separate shared objects. Preconfigured Bazel build configs to DISABLE default on features: --config=noaws # Disable AWS S3 filesystem support. --config=nogcp # Disable GCP support. --config=nohdfs # Disable HDFS support. --config=noignite # Disable Apacha Ignite support. --config=nokafka # Disable Apache Kafka support. --config=nonccl # Disable NVIDIA NCCL support. Configuration finished 编译： hp@dla:~/work/ts_compile/tensorflow$ bazel build --config=opt --config=mkl --verbose_failures //tensorflow/tools/pip_package:build_pip_package 卸载已有tensorflow： hp@dla:~/temp$ sudo pip3 uninstall tensorflow 安装自己编译的成果： hp@dla:~/temp$ sudo pip3 install tensorflow-1.12.0-cp35-cp35m-linux_x86_64.whl

tensorflow基于cpu的检测

10-19

利用python进行tensorflow的检测，亲测有效。基于cpu的操作，接下来就是gpu的

anaconda（Spyder）_tensorflow_cpu/gpu安装配置

01-17

### Anaconda（Spyder）与TensorFlow CPU/GPU 安装配置详解 #### 一、Anaconda环境配置 ##### 1.1 创建Anaconda环境为了确保TensorFlow能够正确安装并运行，在开始之前，首先需要创建一个Anaconda环境。这一步骤...

win10安装cuda10.2+tensorflow-GPU 1.14.0

weixin_45674216的博客

04-14

5836

一、问题提出之前写好的代码都是基于tensorflow 1.14.0的，在自己笔记本上跑太慢了，所以搬到服务器去跑。但是一个很现实的问题就是服务器的CUDA是10.2的，因为服务器有很多人在用，其他人用的是10.2的，而tensorflow 1.14.0对应的CUDA版本是10.0的，我也不太可能去卸载了去重新装一个CUDA，在tensorflow的安装过程中不会出现大的问题，但是在后面运行的时候会报错，会提示找不到：cudart64_100.dll文件，看名称就可以看得出来是64位10.0版本的CUD

tensorflow2使用CUDA10.2

码农ZZK的博客

11-26

2346

tf2是不支持CUDA10.2的，但我们可以做一个软链接，链接到10.1上可参考 [解决]tensorflow-gpu2.3与CUDA10.2的兼容问题 dlerror: libcudart.so.10.1: cannot open shared object file 亲测成功

因安装cuda10.2导致tensorflow1.0级版本（对应cuda10.0）相关代码不兼容的解决方案

weixin_42017945的博客

02-24

1494

因安装cuda10.2导致tensorflow版本相关代码不兼容的解决方案

win10安装cuda10.2+tensorflow 1.14.0

耶律大石的博客

08-27

5332

背景最近为了跑yolov5，我要安装pytorch，然后将cuda 10.0升级到了cuda10.2.pytorch是没问题了，但是我再运行我的yolo v4（tensorflow框架）代码时发现运行不了了，提示：cudart64_100.dll找不到，但是我现在是cuda 10.2了，他还在找cuda10.0。然后网上各种说cuda 10.2没有对应的tensorflow版本，我就觉得再装回去太麻烦了。下面是我的解决办法。重新安装tensorflow-GPU步骤创建环境 conda create

tensorflowcpu和gpu区别

02-14

### TensorFlow CPU与GPU的区别 #### 设计初衷的不同 CPU和GPU由于最初用于处理的任务不同，因此在设计上存在显著区别。某些任务与GPU最初旨在解决的问题较为相似，这使得这些任务更适合由GPU执行[^1]。 #### 版本...