首页 > 编程语言 >Python安装TensorFlow-GPU

Python安装TensorFlow-GPU

时间:2022-12-12 13:25:35浏览次数:80  
标签:Python GPU 版本 gpu tf TensorFlow 安装

选择TensorFlow版本(重要)
验证TensorFlow-gpu安装成功
安装遇到的问题
参考TOC

本文主要介绍windows下基于Miniconda下的GPU版本的TensorFlow安装过程以及安装过程中遇到的问题,本文假设系统下已经成功安装了CUDA和Cudnn,如果未成功安装,请移步另一文章windows11安装CUDA、Cudann

:本文默认基于已经创建好的conda虚拟环境进行安装,所有命令操作均在对应虚拟环境下执行

选择TensorFlow版本(重要)

基于已经安装的CUDA和Cudnn,TensorFlow的版本选择十分重要、十分重要、十分重要(重要事情说三遍)不要一股脑的使用conda install tensorflow-gpu安装,否则无法使用GPU或者后期使用过程中也会存在很多问题。
本人电脑安装的CUDA是11.4版本、虚拟环境中python是3.8.15版本的,因此先去TensorFlow官网查看对应TensorFlow的GPU版本(CPU版本往上扒拉即可),具体如下图:

解释:

  • 查看CUDA对应的版本(是向下兼容的)下可安装的TensorFlow
  • 查看支持的Python版本
  • 因此在红色框住的区域都可以安装,本人选择了2.9.0版本进行安装,具体命令如下:
    conda install tensorflow_gpu-2.9.0

验证TensorFlow-gpu安装成功

在虚拟环境下,进入python环境,输入以下代码依次验证:

import tensorflow as tf
tf.__version__#返回tensorflow-gpu的版本
tf.test.is_gpu_available() #用来验证GPU是否可用
tf.test.is_built_with_cuda() #用来验证cuda是否可用
tf.test.gpu_device_name() #返回gpu的名称
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))#返回可用GPU的个数

出现以下截图,则说明安装成功

安装遇到的问题

  1. 进入python环境时报错,具体错误如下图(忘了保存报错截图了,暂用一个博客内的截图,解决方案也是来自该博客)

    然后按照报错路径找到history.py文件,定位到82行,在82行处添加代码:encoding='utf-8'修改后的代码如下,然后重新执行python成功

参考



来自为知笔记(Wiz)

标签:Python,GPU,版本,gpu,tf,TensorFlow,安装
From: https://www.cnblogs.com/tingstone/p/16975769.html

相关文章

  • Python安装Pytorch-GPU
    选择Pytorch版本(重要)验证pytorch安装是否成功安装遇到的问题参考TOC本文主要介绍windows下基于Miniconda下的GPU版本的Pytorch安装过程以及安装过程中遇到的问题,本文假......
  • Python虚拟环境(二):Linux基于Anaconda创建虚拟环境并打包
    1、创建并查看虚拟环境1、创建虚拟环境condacreate-npy37python3.7#创建一个名称为py37的Python版本为3.7的Python虚拟环境#或condacreate-npy37--copyy......
  • Python虚拟环境(一):基于virtualenv+virtualenvwrapper创建python虚拟环境
    1.概述2.virtualenv+virtualenvwrapper1.virtualenv安装virtualenv基本使用2.virtualenvwrapper安装virtualenvwrapper设置环境变量WORKON_HOMEvirtualenvwrapper的......
  • Go/Python gRPC实践
    gRPC框架&ProtoBuf安装相关工具:pip3installgrpciopip3installgrpcio-toolsprotobuf3有自己专门的定义的格式,基于此可以生成不同的脚本编写示例的protobuf3:......
  • Python 接收解析用户输入参数
    方式一:importsys,getopt;if(__name__=='__main__')or(__name__=='main'):##参数0是文件名+后缀##参数一 Name=sys.argv[1];......
  • tensorflow
    ​​https://github.com/aymericdamien/TensorFlow-Examples.git​​摘要: 利用Docker和阿里云容器服务轻松在本地和云端搭建Tensorflow的学习环境本系列将利用Docker和阿里......
  • 推荐《用Python进行自然语言处理》中文翻译-NLTK配套书
    NLTK配套书《用Python进行自然语言处理》(NaturalLanguageProcessingwithPython)已经出版好几年了,但是国内一直没有翻译的中文版,虽然读英文原版是最好的选择,但是对......
  • Python猫荐书系列:文也深度学习,理也深度学习
    最近出了两件大新闻,相信大家可能有所耳闻。我来当个播报员,给大家转述一下:1、中国队在第11界罗马尼亚数学大师赛(RMM)中无缘金牌。该项赛事是三大国际赛事之一,被誉为中学奥数......
  • python爬虫单线程与多线程区别
    之前有人请我帮忙写一个有关招聘的爬虫,一开始先是单线程,结果显而易见非常慢,后来改了多线程,速度杠杠的。1、单线程importurllibimporturllib.requestimportrequestsimport......
  • Python爬虫实战,requests+openpyxl模块,爬取手机商品信息数据(附源码)
    前言今天给大家介绍的是Python爬取手机商品信息数据,在这里给需要的小伙伴们代码,并且给出一点小心得。首先是爬取之前应该尽可能伪装成浏览器而不被识别出来是爬虫,基本的......