深度学习环境搭建( Tensorflow & PyTorch)

2023-11-11

前言

硬件配置

基础软件 

1,安装VC_redist.x64

2,安装显卡驱动并确定算力

3,确认cuda版本

4,安装CUDA、配置cudnn

5,安装Anaconda 

6,安装PyCharm

深度学习框架Tensorflow安装

深度学习框架PyTorch安装

软件使用

1,使用notebook新建项目

 2,使用PyCharm新建项目

前言

实践是检验真理的唯一标准!

深度学习本身就是一个多学科的领域,需要大量的理论和数学知识使得真正的理解相对困难;而深度学习框架简化了实现的过程,使我们可以相对直观的感受到深度学习的过程,使我们的学习和开发过程事半功倍。

我们以目前最受欢迎的深度学习框架Pytorch & Tensorflow为例,介绍一下环境的搭建。

硬件配置

基础软件 

1,安装VC_redist.x64

安装VC_redist.x64或直接安装vs2019开发环境(这个安装后必须重启电脑

下载链接:https://docs.microsoft.com/zh-CN/cpp/windows/latest-supported-vc-redist?view=msvc-170

2,安装显卡驱动并确定算力

显卡驱动版本,在nvidia控制面板的系统信息可查询到(或者命令行输入nvidia-smi查询),建议更新到最新版本,这样就不会因驱动版本低而下载CUDA版本受限。

NVIDIA官方显卡驱动下载地址:https://www.nvidia.com/Download/index.aspx?lang=en-us

确定算力

3,确认cuda版本

根据显卡算力和驱动版本确定CUDA版本

查询显卡算力对应的CUDA版本区间:https://zh.wikipedia.org/wiki/CUDA

 根据驱动版本进一步确认CUDA版本区间

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

显卡算力和显卡驱动版本共同决定你能下载的CUDA版本区间。

例如:我的电脑显卡是NVIDIA GeForce RTX 3080 Ti,算力是8.6,驱动版本512.15;

根据算力可选CUDA版本应大于11.0,根据驱动版本可选择大于11.0的任意版本。

4,安装CUDA、配置cudnn

安装CUDA时选择自定义安装,对比取消显卡驱动的二次安装NVIDIA GeForce Experience

cudnn下载地址:https://developer.nvidia.com/rdp/cudnn-archive

cudnn无需安装,只需要解压后对应复制到cuda安装目录下(其中include和bin文件夹可以直接移动,lib文件夹内容要移动到lib\x64目录下)

 最后打开cmd,键入nvcc -V验证安装效果

5,安装Anaconda 

 下载路径,默认安装https://www.anaconda.com/products/individual

6,安装PyCharm

 下载路径,默认安装https://www.jetbrains.com/pycharm/

至此,深度学习基本环境搭建OK了,接下来就是针对不同的深度学习框架,在Anaconda中创建对应的虚拟环境来安装了。

下面分为Tensorflow和Pytorch两个开发框架来介绍开发环境的安装!

深度学习框架Tensorflow安装

安装tensorflow-gpu

 

打开anaconda,创建python环境:

conda info --envs        查看现有环境列表

创建Python环境(基于Tensorflow2.6版本,Python3.9版本创建,命令如下)

conda create --name tf2.6 python=3.9

conda activate tf2.6        进入某个环境

pip list        查看当前环境所有的依赖包

conda search --full --name tensorflow        查看所有发现的版本

命令安装Tensorflow2.6(带GPU功能的版本,安装时使用清华镜像源加速)

pip install tensorflow-gpu==2.6.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

或者离线安装:控制台打开cd到已下载的文件目录-- 安装

pip install tensorflow_gpu-2.6.0-cp39-cp39-win_amd64.whl

安装完成后开始测试

anaconda环境中安装附加包

激活环境activate tf2.6再安装附加包

pip install opencv-contrib-python==3.4.1.15 -i https://mirrors.aliyun.com/pypi/simple

pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple/

pip install pandas matplotlib notebook -i https://pypi.tuna.tsinghua.edu.cn/simple/

其余需要的附加包,可在用到时进行安装。

深度学习框架PyTorch安装

创建虚拟环境并配置镜像源

(为了防止冲突,pytorch安装自带cuda runtime。所以不需要下载Cuda单纯下载pytorch就可以使用GPU计算了。)

conda info --envs

conda create -n PT1.11 python=3.9

conda activate PT1.11

配置国内镜像源(提高软件下载速度)

命令行输入:命令行自动配置镜像

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 --set show_channel_urls yes

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

安装PyTorch

打开pytorch官网,选择对应的配置

https://pytorch.org/get-started/locally/#windows-pip

输入下面命令安装

conda install pytorch torchvision torchaudio cudatoolkit=11.3

验证安装效果

安装完成后,测试环境

安装附加包(同Tensorflow)

软件使用

1,使用notebook新建项目

 2,使用PyCharm新建项目

 

 

至此,深度学习框架的开发环境已经搭建完成!

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

深度学习环境搭建( Tensorflow & PyTorch) 的相关文章

随机推荐

  • Java开发环境搭建,保姆级JDK下载,安装,环境变量配置,一系列的完整流程的操作截图以及详细文字说明,JDK,JRE环境变量怎么配?这一篇就够了。

    Java的基础开发环境搭建 前言 提前了解一下以下知识在操作过程中会更加流畅哦 文件目录 https baike baidu com item E6 96 87 E4 BB B6 E7 9B AE E5 BD 95 7204308 fr a
  • MCS51 系列单片机的最小系统

    让单片机工作的前提是为它添加必要的外围电路以构成单片机最小系统 初学者可能对 单片机最小系统感觉很神秘 其实单片机最小系统很简单 就是能使单片机工作的最少的器件构成的系统 最小系统虽然简单 但是却是大多数控制系统所必不可少的关键部分 对于
  • AGG简介

    AGG是一个开源 高效的跨平台2D图形库 AGG的功能与GDI 的功能非常类似 但提供了比GDI 更灵活的编程接口 其产生的图形的质量也非常高 自称超过GDI 使用前AGG的准备工作 下载AGG库 它的家在http www antigrai
  • 5. 使用PyTorch预先训练的模型执行目标检测

    5 使用PyTorch预先训练的网络执行目标检测 PyTorch是什么 使用PyTorch训练第一个神经网络 使用PyTorch训练第一个卷积神经网络 使用预训练网络进行PyTorch图像分类 使用预训练网络进行PyTorch目标检测 今天
  • 记录STM32各类串口中断实验

    STM32F407VG使用cubemx LL库 波特率115200 一位停止位 无校验 单片机初始化串口一 PA8高速推挽输出 使用逻辑分析仪100M采样捕获各类串口中断触发时的波形 首先是喜闻乐见的RXNE 理论是只要串口成功接收一byt
  • Cache 和 Buffer 有什么区别

    Cache 和 Buffer 有什么区别 转载 talkwithtrend https mp weixin qq com s YsEOBVS7fXgrGXnXH1I0MQ Cache和Buffer简单的说 Cache是加速 读 而buffe
  • android自定义可缩放,移动图像裁剪框

    在实际项目中 经常要制作一个简易的图像裁剪功能 即获取一张图片 并用一个遮罩层选择目标范围并截取保存的功能 如下图所示 在此分享下该自定义视图的制作过程 需求说明 整一个视图包含一个透明的遮罩层 一个透明带白色边框的矩形 要实现的功能是 点
  • oracle 问题 :Cause: java.sql.SQLSyntaxErrorException: ORA-00933: SQL 命令未正确结束

    使用mysql时使用的sql语句常有 结尾的语句 但oracle中不能出现 结尾 因为oracle解析器很严谨 不支持 结尾语句解析
  • C基础day6(2023.7.6)

    一 Xmind整理 二 课上练习 练习1 循环嵌套 三个循环结构可以任意嵌套 include
  • Python爬虫分析唯品会商品数据 +数据可视化

    目录 前言 数据来源分析 1 明确需求 2 抓包分析 通过浏览器自带工具 开发者工具 代码实现步骤 发送请求 gt 获取数据 gt 解析数据 gt 保存数据 发送请求 解析数据 保存数据 数据可视化 先读取数据 泳衣商品性别占比 商品品牌分
  • Springmvc+mybatis+Dubbo+ZooKeeper+Redis+KafKa

    开发工具 1 Eclipse IDE 采用Maven项目管理 模块化 2 代码生成 通过界面方式简单配置 自动生成相应代码 目前包括三种生成方式 增删改查 单表 一对多 树结构 生成后的代码如果不需要注意美观程度 生成后即可用 技术选型 只
  • Androidstudio快捷操作之多选

    如图 想要批量复制属性名 不需要一个一个的复制 只需按住alt 鼠标左键从上往下拉 将光标定位在每个属性名的前面 再按ctrl w就可以选中所有光标后面的单词了
  • QT里使用sqlite的问题,好多坑

    1 我使用sqlite 开发机上好好的 测试机上却不行 后来发现是缺少驱动 Driver not loaded Driver not loaded 代码检查了又检查 发现应该是缺少dll文件 系统不提示 是自己使用 QMessageBox
  • c语言基础五子棋,十分的易懂理解,详细解释,容易上手

    前言 提示 经过b站和视频学习后编程 提示 以下是本篇文章正文内容 下面案例可供参考 文章目录 前言 五子棋 头文件 展示棋盘 display 下棋 PlayMove 主体函数 game 完整代码 五子棋 本篇博客主要写了关于c语言的五子棋
  • MySQL主从搭建-Centos实战

    目录 一 规划说明 二 主节点安装MySQL 1 下载MySQL和安装 2 启动Mysql 设置root密码 允许远程登录 三 副节点安装MySQL 参考主节点 四 主节点配置 1 配置my cnf 修改默认存储目录为指定目录 data 下
  • 解决SQL查询总是超时已过期

    解决SQL查询总是超时已过期 在WIN8里提示 OLE DB 或 ODBC 错误 查询超时已过期 HYT00 1 由于数据库设计问题造成SQL数据库新增数据时超时 症状 Microsoft OLE DB Provider for SQL S
  • web前端页面适配方法

    流式布局 就是百分比布局 非固定像素 内容向两侧填充 理解成流动的布局 称为流式布局 视觉窗口 viewport 是移动端特有 这是一个虚拟的区域 承载网页的 承载关系 浏览器 gt viewport gt 网页 适配要求 1 网页宽度必须
  • c++基础十一(跳转语句)

    跳转语句 1 break 2 continue 3 goto 1 break 作用 跳出循环结构和选择结构 1 switch语句中 用于终止case并跳出switch语句 2 在循环结构中 用于跳出当前循环 3 在嵌套循环语句中 跳出最近的
  • 企业微信 => 接入第三方vue应用 第三阶段:企业微信使用JSSDK

    目录 使用说明 官方文档不会告诉你的内容 都是会踩的坑 一 我采用的混入方法去使用这个官方SDK 二 可能会遇到的坑 前提 我们开发的是三方应用 不是内部应用 使用说明 所有的JS接口只能在企业微信应用的可信域名下调用 包括子域名 且可信域
  • 深度学习环境搭建( Tensorflow & PyTorch)

    前言 硬件配置 基础软件 1 安装VC redist x64 2 安装显卡驱动并确定算力 3 确认cuda版本 4 安装CUDA 配置cudnn 5 安装Anaconda 6 安装PyCharm 深度学习框架Tensorflow安装 深度学