神经网络(十五)在vscode下搭建pytorch环境(代码片段)

ViperL1 ViperL1     2023-03-29     794

关键词:

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

一、安装VS Code

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

 二、安装Anaconda

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

python -v

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

 三、下载CUDA工具

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

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

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

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

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

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

nvcc --version

         2.下载CUDnn模块

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

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

三、安装PyTorch

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

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

四、配置VSCode

        1.安装插件

                建议安装以下插件

         2.配置解释器

                在vscode的右下角选择环境

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

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

        3.一点小问题

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

pip list

or

conda list

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

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

最后完美运行

pytorch从头搭建并训练一个神经网络模型(图像分类cnn)(代码片段)

...vision加载数据集并做预处理2.定义(搭建)自己的神经网络3.定义损失函数(LossFunction)和优化器(Optimizer)4.训练神经网络5.测试模型结果6.嫌CPU太慢?换GPU训练并推测试试!0.前言之前用过一些很... 查看详情

PyTorch:简单的前馈神经网络在没有 retain_graph=True 的情况下无法运行

】PyTorch:简单的前馈神经网络在没有retain_graph=True的情况下无法运行【英文标题】:PyTorch:Simplefeedforwardneuralnetworknotrunningwithoutretain_graph=True【发布时间】:2020-06-1506:59:26【问题描述】:以下是我用于训练前馈神经网络(FFNN)的代码... 查看详情

pytorch(代码片段)

...torch是Torch在Python上的衍生物和Tensorflow相比:Pytorch建立的神经网络是动态的,而Tensorflow建立的神经网络是静态的Tensorflow的高度工业化,它的底层代码很难看懂官网:http://pytorch.org/Pytorch主要有两个模块:一个是torch,一个是torchvis... 查看详情

pytorch学习-4:快速搭建+保存提取(代码片段)

快速搭建+保存提取1.快速搭建上一讲神经网络时用到的步骤.我们用net1代表这种方式搭建的神经网络。用class继承了一个torch中的神经网络结构,然后对其进行了修改。classNet(torch.nn.Module):def__init__(self,n_feature,n_hidden,n_output):super(N... 查看详情

pytorch学习-4:快速搭建+保存提取(代码片段)

快速搭建+保存提取1.快速搭建上一讲神经网络时用到的步骤.我们用net1代表这种方式搭建的神经网络。用class继承了一个torch中的神经网络结构,然后对其进行了修改。classNet(torch.nn.Module):def__init__(self,n_feature,n_hidden,n_output):super(N... 查看详情

pytorch神经网络搭建模板(代码片段)

...set:Dataset用于存储数据样本及其对应的标签。在使用神经网络训练时,通常需要将原始数据集转换为Dataset对象,以便能够通过DataLoader进行批量读取数据,同时也可以方便地进行数据增强、数据预处理等操作。Data... 查看详情

pytorch网络搭建中*list的用法解析(代码片段)

问题stage1=nn.Sequential(nn.Sequential(nn.Conv2d(16,32,3,1,1),nn.ReLU(), 查看详情

小白学习pytorch教程十五bert:通过pytorch来创建一个文本分类的bert模型(代码片段)

@Author:Runsen2018年,谷歌发表了一篇题为《Pre-trainingofdeepbidirectionalTransformersforLanguageUnderstanding》的论文。在本文中,介绍了一种称为BERT(带转换器Transformers的双向编码Encoder器表示)的语言模型,该模型在问答、自然语言推理、... 查看详情

vscode搭建docker环境

...云开源镜像站前言本环境通过使用VMware在Ubuntu20.04下通过VScode搭建docker环境,可通过远程连接容器进行开发。(仅作参考)一、环境准备VMware15pro,Ubuntu20.04,VScode(Linux版本)虚拟机搭建 查看详情

pytorch搭建u-net网络(代码片段)

U-Net:ConvolutionalNetworksforBiomedicalImageSegmentationimporttorch.nnasnnimporttorchfromtorchimportautogradfromtorchsummaryimportsummaryclassDoubleConv(nn.Module):def__init__(self,in_ch,out_ch):supe 查看详情

学习pytorch-使用torch搭建一个简单神经网络(代码片段)

a.pyimporttorchimportnumpyasnp#生成一个五行四列的二维矩阵print(torch.Tensor(5,4))#返回的数组大小是5x4的矩阵,初始化是0~1的均匀分布x=torch.rand(5,4)print(torch.randn(5,4))print(x)#查看x的形状print(x.size())#numpy类似的返回5x4大小的矩阵print(np.ones(( 查看详情

在ubuntu下将pytorch模型部署到c++(环境搭建)(代码片段)

...装opencv安装及编译本文总结记录了在将lightweight-openpose的pytorch网络模型部署到c++的过程以及遇到的一些问题,如有错误敬请指正vscodec++环境部署cmakevscode中c++编译过程主 查看详情

《神经网络与pytorch实战》肖智清著部分代码复现与注释,包括使用pytorch搭建cnnrnnlstm等基础神经网络(代码片段)

...研究需要学习的基础章节所以跳过了第七章:全连接神经网络7-1利用torch.nn.Sequential类构建前馈神经网络示例:基于全连接网络的非线性回归第八章:卷积神经网络8-1卷积层用法示例8-2torch.nn.Unsample类用法示例8-3各种类... 查看详情

基于pytorch的神经网络之regression(代码片段)

目录1.引言2.神经网络搭建2.1准备工作2.2搭建网络2.3训练网络3.效果4.完整代码1.引言我们之前已经介绍了神经网络的基本知识,神经网络的主要作用就是预测与分类,现在让我们来搭建第一个用于拟合回归的神经网络吧。2... 查看详情

pytorch深度学习实战|搭建卷积神经网络进行图像分类与图像风格迁移

...#xff0c;对新手非常友好。本文为实战篇,介绍搭建卷积神经网络进行图像分类与图像风格迁移。1、实验数据准备本文中准备使用MIT67数据集,这是一个标准的室内场景检测数据集,一共有67个室内场景,每类包括80... 查看详情

pytorch与深度学习——8.什么是循环神经网络(recurrentneuralnetwork)(代码片段)

在前面的章节里,给大家介绍了什么是神经网络,从基础概念出发,为了识别MNIST数据集,我们搭建了一个全连接网络,然后又把卷积的概念引入进来后,给大家介绍了全连接网络的一种改进型——卷积神... 查看详情

mac下搭建环境homebrew/git/node.js/npm/vscode...

主要记录一下homebrew/git/node.js/npm的命令行安装很简单,下面开始:1.homebrew 也是一个包管理工具:mac里打开终端命令行工具,粘下面一行回车,/usr/bin/ruby-e"$(curl-fsSLhttps://raw.githubusercontent.com/Homebrew/install/master/install)"2.git:同... 查看详情

vscode搭建docker环境(代码片段)

...云开源镜像站前言本环境通过使用VMware在Ubuntu20.04下通过VScode搭建docker环境,可通过远程连接容器进行开发。(仅作参考)一、环境准备VMware15pro,Ubuntu20.04,VScode(Linux版本)虚拟机搭建过程略。二、搭建步骤1.安装docker官方安... 查看详情