admin 管理员组

文章数量: 1184232

windows安装配置pytorch+vscode环境

  • 学习路径
    • 环境
      • 配置目标
      • 环境需求递归
      • 软件安装
      • 调试与配置:

学习路径

环境

配置目标

  1. 为机器学习项目服务-项目需要pytorch库
  2. 适应自己的编程习惯-惯用vscode,我已经安装了python插件

环境需求递归

  1. github项目:ConvLSTM_pytorch->需要pytorch库
  2. pyTorch环境->需要cuda(有GPU情况下)&conda
  3. CUDA(无需手动安装)
    1. GPU为外接1060 3G(穷)

    2. 电脑上安装的是Py3.10(实际上不需要,后面会在虚拟环境中重新安装)

    3. 所以应该选择pytorch-1.12.1-py3.10-cuda11.6的版本 (此处版本考虑并不必要,因为虚拟环境配置时会自动安装)故应选择cuda11.6.2使用

    4. 但anatorch会自动配置cuda环境调用显卡,故不用手动安装。

    5. 所以首先安装conda(anaconda),然后安装PyTorch环境,最后配置。

软件安装

  1. conda-anaconda-软件包/环境管理器
    1. Anaconda官网下载安装
    2. 开始菜单打开Anaconda Prompt
    3. 建立环境
    conda create -n pytorch_env python=3.10     (自己的python版本(并不需要和已经安装的版本相同))
    
    1. 更换下载源:csdn-conda下载换源
    conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/cloud/msys2/
    conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/cloud/conda-forge
    conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/pkgs/free/
    conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/cloud/pytorch/
    conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/pkgs/main/
    conda config --set show_channel_urls yes
    
    1. 进入环境
    conda activate pytorch_env
    
  2. PyTorch环境配置及安装
    1. 从官网找到合适的Pytorch版本并复制指令:pytorch官网

    2. 在刚刚建立好的pytorch_env环境下使用指令

    3. 弹出详情列表,按Y回车安装

调试与配置:

  1. 命令行下:

    成功

  2. vscode下:

    运行失败

文件-首选项-设置-扩展-Python-在settings.json中编辑
在settings.json文件中修改如下:

"python.defaultInterpreterPath": "C:\\Users\\amazc\\AppData\\Local\\Programs\\Python\\Python310\\python.exe",

改为:

本文标签: 入门 环境 Windows pytorch vscode