神经网络(十五)在VS Code下搭建PyTorch环境

这篇具有很好参考价值的文章主要介绍了神经网络(十五)在VS Code下搭建PyTorch环境。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

        本文主要记录如何在电脑中安装pytorch以及在vscode中配置相关环境,以及在配置过程中遇到的一些问题。废话不多说,现在开始。

一、安装VS Code

        前往微软官网下载vs code,选择社区版,默认安装即可

vscode 如何安装pytorch框架,神经网络,学习笔记,环境搭建&故障排除,pytorch,神经网络,python

 二、安装Anaconda

        前往Anaconda官网下载一个安装包,要注意自己电脑的Python版本,可以在控制台中使用

python -v

                查看自己的版本号。我的是3.9所以直接下载最新的

vscode 如何安装pytorch框架,神经网络,学习笔记,环境搭建&故障排除,pytorch,神经网络,python

 三、下载CUDA工具

        1.确定自己的显卡所支持的CUDA版本

                在英伟达控制面板的左下角找到“系统信息”,在“组件”中找到支持的CUDA版本

vscode 如何安装pytorch框架,神经网络,学习笔记,环境搭建&故障排除,pytorch,神经网络,python

                 前往CUDA下载页面,选择自己电脑合适的CUDA版本进行下载。如果在默认页面找不到对应的版本,可以在下面查找历史版本。

vscode 如何安装pytorch框架,神经网络,学习笔记,环境搭建&故障排除,pytorch,神经网络,python

                 安装时可以选择自定义安装,以避免安装全家桶(其实早都装完了),我选择勾掉VS Integration,其他的全部安装。当然,如果C盘内存不够了也可以指定其他盘,并没有什么影响。

vscode 如何安装pytorch框架,神经网络,学习笔记,环境搭建&故障排除,pytorch,神经网络,python

         安装完成后一般会自动部署环境,如果没有自动部署环境的话可以百度查看一下如何部署

        随后在CMD中使用代码验证环境

nvcc --version

vscode 如何安装pytorch框架,神经网络,学习笔记,环境搭建&故障排除,pytorch,神经网络,python

         2.下载CUDnn模块

                CUDnn用于神经网络计算,下载地址为CUDnn下载,完成注册后选择自己对应的CUDA版本进行下载,下载之后,将压缩包中的三个文件夹解压到之前安装CUDA的地方

vscode 如何安装pytorch框架,神经网络,学习笔记,环境搭建&故障排除,pytorch,神经网络,python

                 具体可以参见这篇博文CUDA安装教程

三、安装PyTorch

        前往官网https://pytorch.org/,选择自己对应的情况,获得一句安装口令

vscode 如何安装pytorch框架,神经网络,学习笔记,环境搭建&故障排除,pytorch,神经网络,python

         在Anaconda的控制台中直接粘贴这句话,等待安装成功即可。如果出现done则表示安装成功,国内部分网络可能出现下载过慢/安装失败的情况,可以参考如何解决pytorch下载慢的问题,更换清华镜像来解决。

四、配置VSCode

        1.安装插件

                建议安装以下插件

vscode 如何安装pytorch框架,神经网络,学习笔记,环境搭建&故障排除,pytorch,神经网络,python

         2.配置解释器

                在vscode的右下角选择环境

vscode 如何安装pytorch框架,神经网络,学习笔记,环境搭建&故障排除,pytorch,神经网络,python

                 选择pytorch解释器,没有的话可以到安装路径中查找

vscode 如何安装pytorch框架,神经网络,学习笔记,环境搭建&故障排除,pytorch,神经网络,python

                 如果运行出错的话可以尝试配置一下环境(反正我是没配置,有需要的话可以百度搜索)

        3.一点小问题

                我在完成配置后进行测试发现numpy的版本号与pytorch版本号不一致,可以使用

pip list

or

conda list

                        来核验版本号,如果不对的话记住编译器提示你的正确版本号,在Anaconda的控制台中运行以下代码即可解决

pip uninstall numpy
pip install numpy==1.18.2    --替换为你需要的版本

最后完美运行

vscode 如何安装pytorch框架,神经网络,学习笔记,环境搭建&故障排除,pytorch,神经网络,python文章来源地址https://www.toymoban.com/news/detail-835030.html

到了这里,关于神经网络(十五)在VS Code下搭建PyTorch环境的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

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

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

相关文章

  • 【搭建PyTorch神经网络进行气温预测】

    year month day week temp_2 temp_1 average actual friend 0 2016 1 1 Fri 45 45 45.6 45 29 1 2016 1 2 Sat 44 45 45.7 44 61 2 2016 1 3 Sun 45 44 45.8 41 56 3 2016 1 4 Mon 44 41 45.9 40 53 4 2016 1 5 Tues 41 40 46.0 44 41 数据表中 year,moth,day,week分别表示的具体的时间 temp_2:前天的最高温度值 temp_1:昨天的最高温度值 average:在历

    2024年02月14日
    浏览(51)
  • 【PyTorch框架】——框架安装&使用流程&搭建PyTorch神经网络气温预测

    目录 一、引言 二、使用流程——最简单例子试手 三、分类任务——气温预测   总结: Torch可以当作是能在GPU中计算的矩阵,就是ndarray的GPU版!TensorFlow和PyTorch可以说是当今最流行的框架!PyTorch用起来简单,好用!而TensoFlow用起来没那么自由!caffe比较老,不可处理文本数据

    2024年02月05日
    浏览(42)
  • 小白的机器学习之路(四)神经网络的初步认识:基于pytorch搭建自己的神经网络

    当前交通大数据业务的需要,需要承担一部分算法工作(数据处理) 目标四: 学习深度学习基础:了解神经网络的基本结构、反向传播算法和激活函数等。 目标任务:使用深度学习算法构建一个简单的神经网络模型,并训练模型。 学习计划小贴士: 每天定期复习前几天的

    2024年02月15日
    浏览(39)
  • 人工智能(Pytorch)搭建模型6-使用Pytorch搭建卷积神经网络ResNet模型

    大家好,我是微学AI,今天给大家介绍一下人工智能(Pytorch)搭建模型6-使用Pytorch搭建卷积神经网络ResNet模型,在本文中,我们将学习如何使用PyTorch搭建卷积神经网络ResNet模型,并在生成的假数据上进行训练和测试。本文将涵盖这些内容:ResNet模型简介、ResNet模型结构、生成假

    2024年02月06日
    浏览(78)
  • Pytorch:搭建卷积神经网络完成MNIST分类任务:

    2023.7.18 MNIST百科: MNIST数据集简介与使用_bwqiang的博客-CSDN博客 数据集官网:MNIST handwritten digit database, Yann LeCun, Corinna Cortes and Chris Burges 数据集将按以图片和文件夹名为标签的形式保存:  代码:下载mnist数据集并转还为图片  训练代码: 测试代码: 分类正确率不错:

    2024年02月17日
    浏览(46)
  • 《Pytorch新手入门》第二节-动手搭建神经网络

    参考《深度学习框架PyTorch:入门与实践_陈云(著)》 代码链接:https://github.com/chenyuntc/pytorch-book 神经网络是机器学习中的一种模型,是一种模仿动物神经网络行为特征,进行分布式并行信息处理的算法数学模型。这种网络依靠系统的复杂程度,通过调整内部大量节点之间相互

    2024年02月05日
    浏览(45)
  • 人工智能(pytorch)搭建模型10-pytorch搭建脉冲神经网络(SNN)实现及应用

    大家好,我是微学AI,今天给大家介绍一下人工智能(pytorch)搭建模型10-pytorch搭建脉冲神经网络(SNN)实现及应用,脉冲神经网络(SNN)是一种基于生物神经系统的神经网络模型,它通过模拟神经元之间的电信号传递来实现信息处理。与传统的人工神经网络(ANN)不同,SNN 中的

    2024年02月08日
    浏览(50)
  • Pytorch入门学习——快速搭建神经网络、优化器、梯度计算

    我的代码可以在我的Github找到 GIthub地址 https://github.com/QinghongShao-sqh/Pytorch_Study 因为最近有同学问我如何Nerf入门,这里就简单给出一些我的建议: (1)基本的pytorch,机器学习,深度学习知识,本文介绍的pytorch知识掌握也差不多. 机器学习、深度学习零基础的话B站​吴恩达、

    2024年02月14日
    浏览(42)
  • 学习pytorch13 神经网络-搭建小实战&Sequential的使用

    B站小土堆pytorch视频学习 https://pytorch.org/docs/stable/generated/torch.nn.Sequential.html#torch.nn.Sequential sequential 将模型结构组合起来 以逗号分割,按顺序执行,和compose使用方式类似。 箭头指向部分还需要一层flatten层,展开输入shape为一维 tensorboard 展示图文件, 双击每层网络,可查看层

    2024年02月07日
    浏览(41)
  • 使用自己的数据利用pytorch搭建全连接神经网络进行回归预测

    引入必要的库,包括PyTorch、Pandas等。 这里使用sklearn自带的加利福尼亚房价数据,首次运行会下载数据集,建议下载之后,处理成csv格式单独保存,再重新读取。 后续完整代码中,数据也是采用先下载,单独保存之后,再重新读取的方式。

    2024年02月13日
    浏览(47)

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

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

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

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

二维码1

领取红包

二维码2

领红包