IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 人工智能 -> 【AI】Windows下的环境搭建(Tensorflow-gpu、CUDA&cuDNN、Opencv) -> 正文阅读

[人工智能]【AI】Windows下的环境搭建(Tensorflow-gpu、CUDA&cuDNN、Opencv)

硬件平台:windows10、NVIDIA GeForce GTX 1050

软件平台:Pycharm、Anaconda3、Tensorflows

实现功能 基础平台搭建

一、安装 Pycharm & Ananconda3

两个软件的安装步骤简单。

需注意的是,建议先安装Pycharm,这样在安装完Ananconda后可以直接在Pycharm中创建后续船舶检测分类的虚拟环境。

具体安装及配置步骤,参考:? Anaconda与Pycharm详细安装配置?
?

二、安装CUDA&cuDNN

CUDA是NVIDIA推出的用于自家GPU的并行计算框架,cuDNN(CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。如果你要用GPU训练模型,cuDNN不是必须的,但是一般会采用这个加速库。?

本项目中需要使用GPU进行神经网络的训练,故CUDA 和 cuDNN都需要安装。

需要注意的是,CUDA 和 NVIDIA的版本对应关系是非常重要的,而tensorflow-gpu和CUDA、cuDNN的版本都有严谨的对应关系,版本不对应将无限bug。建议读者先查询tensorflow-gpu的版本对应关系表,再查询CUDA对应NVIDIA驱动版本关系表。

? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?? tensorflow-gpu版本对应关系表

? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?CUDA对应NVIDIA驱动版本关系表

阅表后,按照以下顺序进行安装:显卡驱动更新(去n卡官网即可实现)—— CUDA —— cuDNN 在安装cuDNN时,需注意cuDNN库并不是运行程序,只需要下载解压 cuDNN 文件,并配置 Path 环境变量即可。

具体安装操作步骤,请参考:CUDA & cuDNN

三、安装 tensorflow-gpu

tensorflow-gpu 可以直接在pycharm中使用

pip install tensorflow-gup==你需要的版本 -i https://pypi.mirrors.ustc.edu.cn/simple

特别注意,不要安装tensorflow 否则 会自动有限使用cup版本的神经网络。

安装后可以使用下面的测试代码进行环境测试,可以输出正确的tf-gpu版本即为环境安装成功。

import tensorflow as tf
sess = tf.Session()     #这句运行后会返回GPU的信息,如图3.1所示。若只有cpu信息,则安装失败
a = tf.constant([1,2,3],shape=[3],name='a')
b = tf.constant([1,2,3],shape=[3],name='b')
c=a+b

四、安装opencv?

pip install opencv-python -i https://pypi.mirrors.ustc.edu.cn/simple

总结:笔者的显卡版本是GTX1050的直接更新了497.29版本的驱动;CUDA 11.2 ; cuDNN 8.2.0 ; Tensorflow-gpu 2.6.0 ; python 3.8 ; keras 2.6.0

  人工智能 最新文章
2022吴恩达机器学习课程——第二课(神经网
第十五章 规则学习
FixMatch: Simplifying Semi-Supervised Le
数据挖掘Java——Kmeans算法的实现
大脑皮层的分割方法
【翻译】GPT-3是如何工作的
论文笔记:TEACHTEXT: CrossModal Generaliz
python从零学(六)
详解Python 3.x 导入(import)
【答读者问27】backtrader不支持最新版本的
上一篇文章      下一篇文章      查看所有文章
加:2022-03-03 16:13:20  更:2022-03-03 16:14:18 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/26 18:55:00-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码