Win10配置MMClassification
依赖
- Python 3.8
- CUDA 10.2
- Microsoft Visual C++ 14.0
- PyTorch 1.10.0
- MMCV 1.3.17
- MMClassification 0.17.0
配置CUDA
安装CUDA Toolkit
-
根据需要,下载CUDA Toolkit 10.2,选择本地安装
-
双击exe文件,安装到默认路径下
-
选择自定义安装,除CUDA/Visual Studio Integration外全部选择安装
-
查看系统环境变量是否已经自动配置
-
测试是否安装成功
-
nvcc --version
-
set cuda
-