我的编程空间,编程开发者的网络收藏夹
学习永远不晚

【1】如何安装和卸载tensorflow-CPU和GPU各版本-简单清晰版

短信预约 -IT技能 免费直播动态提醒
省份

北京

  • 北京
  • 上海
  • 天津
  • 重庆
  • 河北
  • 山东
  • 辽宁
  • 黑龙江
  • 吉林
  • 甘肃
  • 青海
  • 河南
  • 江苏
  • 湖北
  • 湖南
  • 江西
  • 浙江
  • 广东
  • 云南
  • 福建
  • 海南
  • 山西
  • 四川
  • 陕西
  • 贵州
  • 安徽
  • 广西
  • 内蒙
  • 西藏
  • 新疆
  • 宁夏
  • 兵团
手机号立即预约

请填写图片验证码后获取短信验证码

看不清楚,换张图片

免费获取短信验证码

【1】如何安装和卸载tensorflow-CPU和GPU各版本-简单清晰版

文章目录


前言

学习随笔,权作记录。

一、tensorflow-cpu指定版本的卸载

>> pip uninstall tensorflow==版本号例如:>> pip uninstall tensorflow==2.1.0

二、tensorflow-gpu指定版本的卸载

>> pip uninstall tensorflow==版本号例如:>> pip uninstall tensorflow-gpu==2.1.0

三、tensorflow-cpu指定版本的安装

1、创建虚拟环境

打开Anaconda prompt(Anaconda3)
在这里插入图片描述
检查当前环境下python的版本

>> python --version

在这里插入图片描述
根据python的版本,创建虚拟环境pachong

>> conda create -n pachong python==3.7.4

按“y”,再按回车即可。
在这里插入图片描述
输入conda info -e,查看全部的环境,可以看到pachong环境已经创建完成。

>> conda info -e

在这里插入图片描述

2、激活虚拟环境

在base环境下输入conda activate pachong,即可进入pachong环境

>> conda activate pachong

在这里插入图片描述

3、安装指定版本的tensorflow

讲镜像源切换为清华大学的镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/cloud/conda-forge/# 设置搜索时显示通道地址conda config --set show_channel_urls yes

再输入,等待安装即可。

>> conda install tensorflow=版本号例如:>> conda install tensorflow=1.15.0

四、tensorflow-gpu指定版本的安装

首先,根据python版本,确定tensorflow-gpu,cuda和cudnn的版本。
在这里插入图片描述
下文以安装tensorflow_gpu-2.1.0为例。

安装CUDA

在我们的具体使用中,其实真正需要的并不是整个CUDA,而是cudatoolkit,所以我们在这里直接安装cudatoolkit
激活虚拟环境conda activate pachong

在这里插入图片描述
输入conda conda install cudatoolkit=10.1,安装10.1版本的cudatoolkit。

在这里插入图片描述
大概是300MB左右。

安装cuDNN

再pachong环境下,输入conda install cudnn=7.6,安装cudnn的7.6版本。

在这里插入图片描述
大概200MB左右。

安装tensorflow_gpu-2.1.0

安装tensorflow使用pip安装,在虚拟环境下输入pip install tensorflow-gpu==2.1.0,即可安装。

在这里插入图片描述

测试tensorflow安装成功

进入虚拟环境tensorflow
在这里插入图片描述
在虚拟环境下,打开python
在这里插入图片描述
在python下输入import tensorflow as tf
在这里插入图片描述
在输入tf.test.is_gpu_available(),即可。若最后为True表示GPU可用,证明已经成功安装Tensorflow的GPU版本。
在这里插入图片描述

可能遇到的问题

如果安装完成,测试的时候出现如下问题:
在这里插入图片描述
解决方案:根据描述的情况将protobuf的版本降到3.19.0以下即可。
具体命令如下:

pip install protobuf==3.19.0

总结

以上就是全部内容,除了换源之外还有类似于直接在命令中加镜像源的下载方式,以及直接下载whl镜像源文件,安装的方式。有兴趣的可以自行搜索。

来源地址:https://blog.csdn.net/SUNYUAN0627/article/details/129348374

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

【1】如何安装和卸载tensorflow-CPU和GPU各版本-简单清晰版

下载Word文档到电脑,方便收藏和打印~

下载Word文档

编程热搜

  • Python 学习之路 - Python
    一、安装Python34Windows在Python官网(https://www.python.org/downloads/)下载安装包并安装。Python的默认安装路径是:C:\Python34配置环境变量:【右键计算机】--》【属性】-
    Python 学习之路 - Python
  • chatgpt的中文全称是什么
    chatgpt的中文全称是生成型预训练变换模型。ChatGPT是什么ChatGPT是美国人工智能研究实验室OpenAI开发的一种全新聊天机器人模型,它能够通过学习和理解人类的语言来进行对话,还能根据聊天的上下文进行互动,并协助人类完成一系列
    chatgpt的中文全称是什么
  • C/C++中extern函数使用详解
  • C/C++可变参数的使用
    可变参数的使用方法远远不止以下几种,不过在C,C++中使用可变参数时要小心,在使用printf()等函数时传入的参数个数一定不能比前面的格式化字符串中的’%’符号个数少,否则会产生访问越界,运气不好的话还会导致程序崩溃
    C/C++可变参数的使用
  • css样式文件该放在哪里
  • php中数组下标必须是连续的吗
  • Python 3 教程
    Python 3 教程 Python 的 3.0 版本,常被称为 Python 3000,或简称 Py3k。相对于 Python 的早期版本,这是一个较大的升级。为了不带入过多的累赘,Python 3.0 在设计的时候没有考虑向下兼容。 Python
    Python 3 教程
  • Python pip包管理
    一、前言    在Python中, 安装第三方模块是通过 setuptools 这个工具完成的。 Python有两个封装了 setuptools的包管理工具: easy_install  和  pip , 目前官方推荐使用 pip。    
    Python pip包管理
  • ubuntu如何重新编译内核
  • 改善Java代码之慎用java动态编译

目录