卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」一,卸载CPU版本,如下图之前我已经安装了anaconda,现在检查它的版本以及环境。发现有tensorflow环境,但我需要tensorflow-gpu的环境,所以输入condauninstalltensorflow卸载了,如果发现卸不掉就anaconda页面中点enviroments,选择tensorflow点击下面的remove就可以卸载干净。再次输入condain…

大家好,又见面了,我是你们的朋友全栈君。

一,卸载CPU版本,如下图

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

之前我已经安装了anaconda,现在检查它的版本以及环境。发现有tensorflow环境,但我需要tensorflow-gpu的环境,所以输入conda uninstall tensorflow卸载了,

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

如果发现卸不掉就anaconda页面中点enviroments,选择tensorflow点击下面的remove就可以卸载干净。

再次输入conda info –envs,结果如下图所示:

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

输入conda creat -n tensorflow-gpu pip python=3.6安装tensorflow-gpu环境,

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

再次检查可以看出我们成功创建tensorlfow-gpu环境:

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

二,安装CUDA和CuDNN

1.查看自己的显卡

只有NVIDIA显卡才支持用GPU跑TensorFlow,查询GPU是否支持CUDA,一般要计算能力在3.0以上才适合。

(查看链接:https://developer.nvidia.com/cuda-gpus

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

2.安装显卡驱动

显卡硬件明确之后,安装显卡的驱动,链接http://www.geforce.cn/drivers

我的是1060,就是10series系列,下拉可以找到。我选择2019/9/10发布的。

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

下载完毕后,安装,选项默认不要随便改。

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

3.安装CUDA

首先看看tensorflow和CUDA以及 cuDNN对应的版本。本人用了tensorflow-gpu1.12+CUDA9.0+cuDNN7.0

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

下载完毕是cuda_9.0.176_win10.exe。然后就是双击安装,选择自定义,在选择安装项时一般不安装GeForce Experience,CUDA是核心组件必须勾上,接着点击下一步就开始安装了。

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

在安装的过程中发现安装出现错误,将CUDA选项下面的VS勾掉就可以,如下图:

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

4.安装cuDNN

cuDNN就相当于cuDA显卡计算库中专门针对Depth Neural Network深度神经网络的计算专用库。

cuDNN的下载地址:https://developer.nvidia.com/rdp/cudnn-download。需要注册比较简单。

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

下载完毕后解压缩,得到三个文件夹,将这三个文件拷贝到~CUDA\v9.0这个目录下,集替换。

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

三,安装tensorflow1.12.0

在第一步我们已经创建了tensorlfow-gpu环境,现在激活并进入环境中:

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

安装tensorflow输入语句:pip install –ignore-installed –upgrade tensorflow-gpu==1.12.0

卸载tensorflow的CPU版本并安装GPU版本「建议收藏」

验证成功安装

(tensorflow-gpu) C:\Users\2018061801>python
Python 3.6.9 |Anaconda, Inc.| (default, Jul 30 2019, 14:00:49) [MSC v.1915 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> tf.__version__
'1.12.0'
>>>

 

 

 

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请联系我们举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/152132.html原文链接:https://javaforall.net

(0)
全栈程序员-站长的头像全栈程序员-站长


相关推荐

  • 20道经典Redis面试题

    20道经典Redis面试题前言整理了 20 道经典 Redis 面试题 希望对大家有帮助 1 什么是 Redis 它主要用来什么的 Redis 英文全称是 RemoteDictio 远程字典服务 是一个开源的使用 ANSIC 语言编写 支持网络 可基于内存亦可持久化的日志型 Key Value 数据库 并提供多种语言的 API 与 MySQL 数据库不同的是 Redis 的数据是存在内存中的 它的读写速度非常快 每秒可以处理超过 10 万次读写操作 因此 redis 被广泛应用于缓存 另外 Redis 也经常用来做分布式锁

    2025年8月31日
    3
  • unity安装后续添加模块

    unity安装后续添加模块unitybuildsettingtargetplatform缺少一些选项(linux,mac)unity官方解决方案如下但是unityhub没有addmodule选项前往unity官网下载,找到对应unity版本,点击Releasenotes即可下载对应安装包

    2022年6月27日
    32
  • setUp()和tearDown()函数

    setUp()和tearDown()函数

    2022年4月2日
    90
  • 小程序直播开发文档_小程序直播平台

    小程序直播开发文档_小程序直播平台参考文档:https://www.cnblogs.com/pass245939319/p/9001137.html今天研究了新东西小程序的直播开发这个东西我研究了整整一下午快给我整哭了我的demo地址:https://github.com/AloneYan/WeChat-live.git下面进入正题首先你必须搞一个小程序公众平台的账号注册一个自己的小程序获得自己的…

    2022年9月26日
    3
  • react 纯函数组件_react类组件

    react 纯函数组件_react类组件纯函数PureFunction定义:一个函数的返回结果只依赖于它的参数,并且在执行的过程中没有副作用,我们就把该函数称作纯函数。特点1.函数的返回结果只依赖于它的参数。letfoo=(a,b)=>a+bfoo(1,2)//=32.函数执行过程里面没有副作用。什么是副作用除了修改外部的变量,一个函数在执行过程中还有很多方式产生外部可观察的变化,比如说调用DOMAPI修改页面,或者你发送了Ajax请求,还有调用window.reload刷新浏览器,

    2025年7月4日
    2
  • Pytest(11)allure报告[通俗易懂]

    Pytest(11)allure报告[通俗易懂]前言allure是一个report框架,支持java的Junit/testng等框架,当然也可以支持python的pytest框架,也可以集成到Jenkins上展示高大上的报告界面。mac环境:

    2022年7月30日
    43

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号