栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

win10配置MMClassification+PyTorch+CUDA

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

win10配置MMClassification+PyTorch+CUDA

Win10配置MMClassification

依赖

  1. Python 3.8
  2. CUDA 10.2
  3. Microsoft Visual C++ 14.0
  4. PyTorch 1.10.0
  5. MMCV 1.3.17
  6. MMClassification 0.17.0
配置CUDA 安装CUDA Toolkit
  1. 根据需要,下载CUDA Toolkit 10.2,选择本地安装

  2. 双击exe文件,安装到默认路径下

  3. 选择自定义安装,除CUDA/Visual Studio Integration外全部选择安装

  4. 查看系统环境变量是否已经自动配置

  5. 测试是否安装成功

    1. nvcc --version

    2. set cuda

安装cuDNN
  1. 下载对应的cuDNN,这里需要登录自己的NVIDIA账号

  2. 解压下载的cuDNN文件,将解压后的文件复制到CUDA的安装目录下

  3. 添加系统变量,在path中添加以下路径(根据自己的安装目录进行修改)

    C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2bin
    
    C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2include
    
    C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2lib
    
    C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2libnvvp
    
  4. 测试是否安装成功,在安装目录的extrasdemo_suit下分别执行bandwidthTest.exe和deviceQuery.exe

    [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-hjaepncf-1637562261229)(https://i.loli.net/2021/11/22/RmXP7b1wBgWFJzT.png)]

配置Microsoft Visual C++ 14.0
  1. 下载Microsoft Visual C++ Build Tools

  2. 只需选择C++的桌面开发,然后安装即可

    [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-CQRzXhqv-1637562261230)(https://i.loli.net/2021/11/22/nIQjl2igVSMeWOG.png)]

配置PyTorch
  1. 进入PyTorch官网

  2. 选择1.10版本,CUDA 10.2

  3. 安装:conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

  4. 验证

配置MMCV

这里我们需要的是MMCV-full而不是mmcv

  1. 进入mmcv网站

  2. 选择cuda10.2 + torch1.10 + mmcv 1.3.17

  3. 安装:pip install mmcv-full==1.3.17 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10.0/index.html

  4. 验证

配置MMClassification
  1. 进入MMClassification网站

  2. 下载项目

    git clone https://github.com/open-mmlab/mmclassification.git
    cd mmclassification
    pip install -e .
    
  3. 验证

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/580827.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号