| |
|
开发:
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中使用
特别注意,不要安装tensorflow 否则 会自动有限使用cup版本的神经网络。 安装后可以使用下面的测试代码进行环境测试,可以输出正确的tf-gpu版本即为环境安装成功。
四、安装opencv?
总结:笔者的显卡版本是GTX1050的直接更新了497.29版本的驱动;CUDA 11.2 ; cuDNN 8.2.0 ; Tensorflow-gpu 2.6.0 ; python 3.8 ; keras 2.6.0 |
|
|
上一篇文章 下一篇文章 查看所有文章 |
|
开发:
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- |
|
网站联系: qq:121756557 email:121756557@qq.com IT数码 |