博主安装路径:
conda环境:W:\envs
openmmlab:"W:\coding resources"

安装 PyTorch 和 OpenMMLab 教程

环境准备
  1. 安装 Anaconda

  2. 安装 Git

    • 访问 Git官网 下载并安装 Git。在安装过程中选择目标路径为 W:\coding resources
创建 Conda 环境
  1. 打开 Anaconda Prompt

    • 在开始菜单搜索“Anaconda Prompt”并打开。
  2. 创建新的 Conda 环境

    • 输入以下命令,创建名为 openmmlab 的环境,使用 Python 3.8:
      conda create -p W:\envs\openmmlab python=3.8
      
  3. 激活环境

    • 输入以下命令激活环境:
      conda activate W:\envs\openmmlab
      
安装 PyTorch
  1. 确定 CUDA 版本

    • 确认你的 CUDA 版本,输入以下命令:
      nvcc --version
      
    • 根据你的输出,你的 CUDA 版本是 12.1。
  2. 安装 PyTorch

    • 在激活的 Conda 环境中输入以下命令,安装与 CUDA 12.1 兼容的 PyTorch 版本:
      conda install pytorch torchvision torchaudio cudatoolkit=12.1 -c pytorch
      
  3. 验证 PyTorch 安装

    • 输入以下命令检查 PyTorch 是否正确安装:
      python -c "import torch; print(torch.__version__)"
      
    • 如果没有错误,并且打印出版本号,说明安装成功。
克隆 OpenMMLab 项目
  1. 使用 Git 克隆项目

    • 在 Anaconda Prompt 中,输入以下命令克隆 OpenMMLab 项目到指定路径:
      git clone https://github.com/open-mmlab/mmclassification.git W:\coding resources\mmclassification
      
    • 进入项目目录:
      cd W:\coding resources\mmclassification
      
  2. 安装项目依赖

    • 在项目目录中,输入以下命令安装依赖:
      pip install -r requirements.txt
      
Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐