从0手把手带你搭建pytorch深度学习

这篇具有很好参考价值的文章主要介绍了从0手把手带你搭建pytorch深度学习。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

目录

一、查看电脑有NVIDIA显卡没

二、更新电脑驱动

三、安装CUDA ToolKit和CUDNN

1、查看显卡驱动版本

2、查看合适的CUDA版本

3、下载CUDA ToolKit

4、安装CUDA

5、查看是否安装成功

6、安装CUDNN

7、CUDNN配置

四、安装anaconda

五、安装pycharm

六、搭建pytorch深度学习环境

1、进入Anaconda Prompt(鼠标左击win标志,去找)

2、下载torchvision和torch离线版本(因为在线装有时候会被中断)

3、离线安装

七、搭建pycharm环境测试


一些必要的介绍(自己的理解,有错请大神指教):

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

下面开始啦!!!

一、查看电脑有NVIDIA显卡没

特别是笔记本的,必须要有NVIDIA的独立显卡。比如我的笔记本,打开任务管理器,如下,就有一个GeForce 920M的独立显卡。

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

如果没有,就只能装CPU版的pytorch,当然不利于搞深度学习。不过现在的笔记本一般都有,搞深度学习当然最好用台式机啦,然后配高级点的显卡,至少1080。

二、更新电脑驱动

一些 “废话”:

NVIDIA图形驱动程序是用来驱动NVIDIA显卡的程序,是硬件所对应的软件,用于提升操作系统对其芯片组的兼容性。驱动程序即添加到操作系统中的一小块代码,其中包含有关硬件设备的信息。

有了此信息,计算机就可以与设备进行通信。驱动程序是硬件厂商根据操作系统编写的配置文件,可以说没有驱动程序,计算机中的硬件就无法工作。

操作系统不同,硬件的驱动程序也不同,各个硬件厂商为了保证硬件的兼容性及增强硬件的功能会不断地升级驱动程序。

这一步容易忽略,比较重要。一定要去下面这个网址(这里根据你电脑的显卡型号选择出来的驱动版本是满足你显卡的最高版本的驱动,如果显卡型号选错了,下错驱动版本,是装不上的):

Official Drivers | NVIDIA

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

然后下载安装就行。安装好之后,打开cmd窗口,键入nvidia-smi命令,结果如下:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

这就是安装之后的驱动版本信息,后续会根据这个信息选择CUDA ToolKit + CUDNN的安装。

三、安装CUDA ToolKit和CUDNN

1、查看显卡驱动版本

打开cmd窗口,键入nvidia-smi命令,如下:

 搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

2、查看合适的CUDA版本

CUDA 12.0 Update 1 Release Notes

去这个网站找最低版本的CUDA,如下(注意你的系统是linux还是windows)

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

我们的驱动是425.31,那么找满足大于等于425.31的最低版本CUDA,只有CUDA 10.1了。

3、下载CUDA ToolKit

CUDA Toolkit Archive | NVIDIA Developer

去这个地方下载对应版本的CUDA。此处有三个10.1的,我们下载最新的这个

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

如下根据自己的情况选择:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

4、安装CUDA

双击下载好的.exe文件(我们这里下载的是network版):

第一次出现的临时解压路径,默认就好了,安装结束后,临时解压文件夹会自动删除;

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

下面出现两种选择,精简(下载好所有组件,并且会覆盖原有驱动),我们这里就选择精简吧。如果是自定义,需要选择安装路径,也可以默认。

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

出现如下:说不支持vs,我们选择I understand,不管它,继续。

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

5、查看是否安装成功

cmd窗口键入nvcc --version,如下,出现我们安装的版本,成功。

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

6、安装CUDNN

cuDNN Archive | NVIDIA Developer

从这网址进入下载与CUDA ToolKit匹配的CUDNN,选择一个满足要求最新的:

 搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

点击后我们选择这个:windows 10

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

 进去之后,要求注册邮箱,输入即可:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

 如果之前注册过,忘记密码了,可以选择登录帮助:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

7、CUDNN配置

解压下载好的.zip文件,得到如下:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

将这三个 文件夹下的文件 拷贝到CUDA ToolKit安装路径 对应文件夹下去 ,记不到安装路径,cmd窗口打开键入set cuda,如下:
搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

到这里CUDA ToolKit和CUDNN就弄好了。

四、安装anaconda

这个肯定简单,但是请看到最后,可以安装一个不带界面的精简版,很小。

Anaconda | Start Coding Immediately

进入这个官网,点击右上角的free-download。然后选择对应版本。

上面下载的是有界面的,这里再给一个mini版的链接,没有界面,只有命令行的版本:

Miniconda — conda documentation

下面进去之后,左侧选择自己想要的版本,我这里选择windows,右边的python版本是,你安装anaconda的时候,会自动创建一个base虚拟环境,这个里面的python版本就即你选择的python版本。下载好之后,双击.exe按照步骤安装即可。

五、安装pycharm

Download PyCharm: Python IDE for Professional Developers by JetBrains

进入上面的官网,一般选择社区版本即可:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

 下载好之后双击.exe按照步骤安装即可。

六、搭建pytorch深度学习环境

1、进入Anaconda Prompt(鼠标左击win标志,去找)

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

根据命令:conda create --name xxx python=xxx

新建一个虚拟环境,我这里如下:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

 然后键入y,回车,如下:

 搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

他会自动下载一些必要的库。

2、下载torchvision和torch离线版本(因为在线装有时候会被中断)

torch与torchvision的版本对应可以从如下这个网址里面查:

Previous PyTorch Versions | PyTorch

(同样的CUDA版本,该网页越上面,torch和torchvision的版本越高。但是也越大,于是我这里折中,选个中间的)

比如,我们这里就选择的这个

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

 注意虚拟环境要满足组各种包对应cuda≥系统的CUDA版本

开始下载,进入下面的网页:

https://download.pytorch.org/whl/torch_stable.html

找到对应的版本(上面装的CUDA ToolKit10.1),那么我们选择如下:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

按照同样规则,选出torch:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

3、离线安装

下载好之后,我把它们放到某个文件夹中,如下:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

打开Anaconda Prompt窗口,进入创建的好的虚拟环境,

再cd到这个文件夹,如下:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

然后分两次键入pip install torch-xxxx命令和pip install torchvision-xxx,如下是我的:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

pip list查看安装成功了:

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

七、搭建pycharm环境测试

桌面新建一个文件夹,打开后,设置项目解释器为我们新建的虚拟环境,在main.py中键入以下代码,运行。

import torch

print(torch.__version__)

print('gpu', torch.cuda.is_available())

出现如下结果,恭喜你成功了!

搭建pytorch,配置安装教程,深度学习,pytorch,人工智能

注:如果第二点那里你想安装其他的版本的驱动,可以去这个网址找:

游戏常用软件下载_显卡驱动下载 _ 游民星空 Gamersky.com

最后,给自己鼓个掌吧👏👏👏,坚持到这里也不容易。有任何疑虑,请评论区留言,鄙人水平有限,若有误请赐教。文章来源地址https://www.toymoban.com/news/detail-728635.html

到了这里,关于从0手把手带你搭建pytorch深度学习的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • YOLOv5入门实践(1)——手把手带你环境配置搭建

      这两天我将pycharm社区版换成了专业版,也顺带着把环境从CPU改成了GPU版,本篇文章也就是我个人配置过程的一个简单记录,希望能够帮到大家啦~ ​   🍀本人YOLOv5源码详解系列:  

    2024年02月04日
    浏览(21)
  • 手把手带你搭建Linux环境(保姆级,包含vivado与petalinux)

    目录 1.安装VMware Workstation 2.安装ubuntu, 3.安装vivado和sdk 4.安装petalinux VMware和ubuntu尽量从官网下载自己所需要的版本,避免之后应用时后会有很多工具包不支持,官网下载速度都很快。 提醒各位注意版本控制!! 本文建议安装环境为VMware16pro版本,ubuntu16.04版本,vivado2018.3版本

    2023年04月08日
    浏览(22)
  • Pytorch:手把手教你搭建简单的全连接网络

     红色的点就是我在sinx函数上取的已知点作为网络的训练点。  训练过程如上,时间我这里设置的比较简单,除了分钟,之后的时间没有按照60进制规定。 可以看到收敛的还是比较好的。 这里红色的点为训练用的数据,蓝色为我们的预测曲线,可以看到整体上拟合的是比较好

    2024年02月07日
    浏览(21)
  • 失眠大数据专家,手把手带你玩转大数据,HDFS三种搭建方式

    (1) 配置免密登录 node01-node01 (2) 配置JDK (3) 修改hdfs-site.xml配置文件 (4) 修改core-site.xml配置文件 (5) 修改slaves配置文件 修改为node01 (6) 格式化NameNode(创建目录以及文件) hdfs namenode -format (7) 启动HDFS start-dfs.sh (8) 操作HDFS文件系统 ① 创建目录 hdfs dfs -mkdir -p /user/root ② 上传文件 hdf

    2024年04月11日
    浏览(14)
  • 手把手带你学习 JavaScript 的 ES6 ~ ESn

    JavaScript 是一种广泛使用的网络编程语言,它在前端开发中扮演着重要角色。随着时间的推移,JavaScript 的版本不断更新,新的功能和语法不断涌现,使得这门语言更加丰富和强大。ES6~ESn 是指 JavaScript 的版本6到版本n(例如ES7、ES8等),这些新版本为我们带来了许多新的特性

    2024年02月05日
    浏览(19)
  • 手把手带你用Python和文心一言搭建《AI看图写诗》网页项目(附上完整项目源码)

    今年年初,ChatGPT的火爆在全球掀起AI大模型的开发热潮,国内外的科技公司纷纷加入“百模大战”行列。百度在率先发布了国内第一款人工智能大语言模型“文心一言”后,又推出了文心千帆大模型平台,帮助企业和开发者加速大模型应用落地。 而最近百度创始人、董事长兼

    2024年02月08日
    浏览(15)
  • 手把手带你学习微信小程序 —— (微信小程序常用代码块)

    line-heigth: 50rpx; 7.设置分割线 border-bottom: 1px solid #e4e4e4; 8. 居中布局的一种方式 margin: 0 auto; 9. 垂直居中 三、微信官方 API ======================================================================== 3.1 普通接口 3.1.1 修改当前界面的标题 wx.setNavigationBarTitle({ title: “”, }) 3.1.2 加载响应,。。。 wx

    2024年04月15日
    浏览(23)
  • 【第二趴】uni-app开发工具(手把手带你安装HBuilderX、搭建第一个多端项目初体验)

    聚沙成塔——每天进步一点点,大家好我是几何心凉,不难发现越来越多的前端招聘JD中都加入了uni-app 这一项,它也已经成为前端开发者不可或缺的一项技能了,所以凉哥为大家推出 聚沙成塔【45天玩转uni-app】 专栏,帮助大家对 uni-app 进行学习和理解,uni-app可以通过一套代

    2023年04月23日
    浏览(67)
  • 【深度学习】手把手教你使用 PyCharm 连接 Auto DL 远程服务器

    文章性质: 实操记录 💻 主要内容: 主要记录了如何租用 Auto DL 服务器,以及如何在 PyCharm 中连接远程服务器。 相关文档: 如何使用 Auto DL 远程服务器连接 PyCharm 运行代码 - 知乎 冷知识+1: 小伙伴们不经意的 点赞 👍🏻 与 收藏 ✨ 可以让作者更有创作动力!  目录 前言

    2024年04月25日
    浏览(38)
  • 【毕设必备】手把手带你用Python搭建一个简单的后端服务- API的创建,前后端交互的数据传递,GET,POST,JSON,FLASK

    Python是一种 流行 的高级编程语言,具有易于学习和使用的特性,被广泛应用于各种领域。 简单易学 :Python的语法清晰简洁,易于理解和学习。与其他编程语言相比,Python的语法设计非常直观,使得编程新手也能快速上手。 强大的标准库和丰富的第三方库 :Python拥有一个庞

    2024年02月04日
    浏览(24)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包