码农知识堂 - 1000bd
  •   Python
  •   PHP
  •   JS/TS
  •   JAVA
  •   C/C++
  •   C#
  •   GO
  •   Kotlin
  •   Swift
  • 搭建深度学习环境(Pytorch)


    1 配置问题

    (1).针对于电脑中配备有GPU,且有深度学习需求,搭建一个可用无污染的深度学习环境。

    (2).按照Anaconda3 + CUDA10.0 + CuDNN + Pytorch1.2 + Pycharm配置。

    (3).解决配置Pytorch中无法使用torchvision库的问题。

    2 搭建过程

    (1).安装Anaconda。按照自己电脑的需求下载,下载完成后,解压安装包,按照指示一步一步完成即可,检验安装是否成功,并创建一个虚拟环境用于之后的配置。

    # 创建虚拟环境:(在cmd终端输入)
    conda create -n Pytorch_GPU python=3.7

    bf381943139ecf50dd3ab13b1dfe5921.png

    (2).安装CUDA10.0和cuDNN(必须与自己的Pytorch==1.2.0版本对应起来)

    1. .在官网找到合适的版本进行下载并进行安装。
    2. .找到安装的位置对cuDNN进行解压。
      11e7f74043b18454d10f98ed81fa18b2.jpeg
      0a6bada7fb87a606a23603f23a8dce69.png
      (3).安装Pytorch。
      1. . 激活并进入Anaconda提前准备好的虚拟环境。

    activate Pytorch_GPU

    2).下载安装Pytorch。打开pytorch的官网,地址:https://pytorch.org/。

    也可以去下载轮子(https://download.pytorch.org/whl/torch_stable.html)

    a96d83e3edaf4597ad3a0b591927a4e1.png

    023d7ce379c5d4832709606e0ad5be28.png

    (4)Pycharm和Anaconda连接。

    1. .设置相关环境,网上有很多教程。
      2849f98742f0c5a6603d606cf935edf3.png
    2. .检测Pytorch配置成功。
      a54e53bf453545a5371513335c0d50e3.png
      (5).torchvision问题。
      在导入torchvision的时候出现了报错,但是查询发现torchvision包在库里。
      在网上百度找到问题的解决办法。
      c21cf57d5f69c6e81f3dde632bed2c7f.png
      找到报错路径C:\Users\Public\Anaconda3\Lib\site-packages\PIL,打开__init__.py文件,使用from PIL import Image, ImageOps, ImageEnhance, __version__ 替换文件中from PIL import Image, ImageOps, ImageEnhance,PILLOW_VERSION这句。
    3. 结语

    1.要注意各个软件版本之间的匹配问题。

    2.最新版本的CUDA不再需要配置Cudnn进行加速了。

    3.针对torchvision导入模块缺失问题,更改__init__.py文件的导入模块名,就可以使用了。

  • 相关阅读:
    SpringBoot+Vue实现前后端分离的家政服务平台
    三道动态规划题-最长的有效括号、组合总和I、组合总和II
    100天精通Python(可视化篇)——第100天:Pyecharts绘制多种炫酷漏斗图参数说明+代码实战
    8086 汇编笔记(三):第一个程序
    C语言-指针详解速成
    Bootstrap响应式轮播效果网页(1+X Web前端开发中级 例题)
    工厂车间安灯呼叫系统实现生产过程的可视化管理
    哈希表学习笔记
    自制操作系统日志——第十二天
    浅谈C++|构造.析构函数篇
  • 原文地址:https://blog.csdn.net/gschen_cn/article/details/126912993
  • 最新文章
  • 【JVM】编译执行与解释执行的区别是什么?JVM 使用哪种方式?
    用 Hashids 优雅解决 C 端自增 ID 暴露问题
    V8引擎 精品漫游指南--Ignition篇(上) 指令 栈帧 槽位 调用约定 内存布局 基础内容
    LLVM Pass快速入门(四):代码插桩
    milkup:桌面端 markdown AI续写和即时渲染
    基于项目工程构建SBOM(软件物料清单)的研究
    鸿蒙应用开发UI基础第二节:鸿蒙应用程序框架核心解析与实操
    .NET 中如何快速实现 List 集合去重?
    扣子Coze实战:从0到1打造抖音+小红书热点监控智能体
    浅谈数据访问层
  • 热门文章
  • 十款代码表白小特效 一个比一个浪漫 赶紧收藏起来吧!!!
    奉劝各位学弟学妹们,该打造你的技术影响力了!
    五年了,我在 CSDN 的两个一百万。
    Java俄罗斯方块,老程序员花了一个周末,连接中学年代!
    面试官都震惊,你这网络基础可以啊!
    你真的会用百度吗?我不信 — 那些不为人知的搜索引擎语法
    心情不好的时候,用 Python 画棵樱花树送给自己吧
    通宵一晚做出来的一款类似CS的第一人称射击游戏Demo!原来做游戏也不是很难,连憨憨学妹都学会了!
    13 万字 C 语言从入门到精通保姆级教程2021 年版
    10行代码集2000张美女图,Python爬虫120例,再上征途
小工具 小游戏
Copyright © 2022 侵权请联系2656653265@qq.com    京ICP备2022015340号-1

京公网安备 11010502049817号