这里写自定义目录标题
环境记录
- 笔记本: MSI GE63 Raider RGB 8SG-040CN
- CPU: i7-8750H
- GPU: RTX 2080
- 系统: Win10 专业版
- 显卡驱动: 496.76
- Python: 3.7.13
- Pytorch: 1.10.1
- CUDA: 10.2
- cuDNN: 8.3.1
- MMCV: 1.5.1
- mmdetection: 2.24.1
Step1: 环境配置
安装Anaconda, CUDA, cuDNN
教程很多,不细说了,假设已经能跑成功PyTorch代码了。
创建Conda环境
conda create --name mmdet python=3.7
conda activate mmdet
安装PyTorch
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch
查看CUDA
版本:
nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:32:27_Pacific_Daylight_Time_2019
Cuda compilation tools, release 10.2, V10.2.89
查看cuDNN
版本:
进入C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include
找到cudnn_version.h
打开就可以看到
#ifndef CUDNN_VERSION_H_
#define CUDNN_VERSION_H_
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 3
#define CUDNN_PATCHLEVEL 1
cuDNN
版本即为8.3.1
安装MSVC编译器
如果是CUDA10.2
的话,这里只能安装Visual Studio 2015
到Visual Studio 2019
之间的版本,我这里是安装的Visual Studio 2019
下载链接 安装Community
版就行。
如果是CUDA10.2
的话
Visual Studio 2022 亲测不行,后面编译会报错!!!
Visual Studio 2022 亲测不行,后面编译会报错!!!
Visual Studio 2022 亲测不行,后面编译会报错!!!
安装好后配置环境变量:
找到Visual Studio
安装文件夹,我这里是在E盘安装的:
找到并进入VC
文件夹
找到并进入Tools
文件夹:
进入MSVC
文件夹
进入14.29.30133
文件夹,这里因版本而异,数字可能不一样,点进去就行
进入bin
文件夹
进入Hostx86
文件夹