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


            本文主要记录如何在电脑中安装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版本号不一致,可以使用

    1. pip list
    2. or
    3. conda list

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

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

    最后完美运行

  • 相关阅读:
    4、FFmpeg命令行操作9
    LINUX下看门狗的使用
    css的概念+书写位置+CSS样式规则与应用案例
    【树莓派】Scrot截图工具
    linux修改用户密码脚本
    Springboot楼盘销售系统hze57计算机毕业设计-课程设计-期末作业-毕设程序代做
    Spark任务调度概述_大数据培训
    【架构】酒店预订应用程序的系统设计架构
    AX=0和AX=b的解向量线性相关吗?
    Oracle数据库字符集概述及修改方式
  • 原文地址:https://blog.csdn.net/weixin_37878740/article/details/127069779