神经网络（十五）在VS Code下搭建PyTorch环境

8月前作者：ViperL1 分类：Toy博客阅读(37) 违法举报

这篇具有很好参考价值的文章主要介绍了神经网络（十五）在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日
浏览(41)
【PyTorch框架】——框架安装&使用流程&搭建PyTorch神经网络气温预测

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

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

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

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

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

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

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

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

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

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

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

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

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

2024年02月14日
浏览(34)
学习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日
浏览(33)
人工智能(Pytorch)搭建模型1-卷积神经网络实现简单图像分类

本文参加新星计划人工智能(Pytorch)赛道：https://bbs.csdn.net/topics/613989052 目录一、Pytorch深度学习框架二、卷积神经网络三、代码实战内容：一、Pytorch深度学习框架 PyTorch是一个开源的深度学习框架，它基于Torch进行了重新实现，主要支持GPU加速计算，同时也可以在CPU上运行

2024年02月03日
浏览(50)