环境配置
- 根据硬件确定本地CUDA版本
- 查找对应驱动版本
- 安装合适的pytorch-gpu版本
Anaconda+pytorch+cudatoolkit
- Anaconda安装,直接清华镜像下载更快
#添加镜像源
channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
ssl_verify: true
#显示检索路径
conda config --set show_channel_urls yes
#显示镜像通道
conda config --show channels
- pytorch-gpu
step1: conda create -n pytorch python=3.8
step2: conda install pytorch=1.11.0 cudatoolkit=11.2 torchvision
step3: 安装其他需要依赖库
框架mmdet
因为mmdet需要从本地CUDA编译,而不是使用pytorch安装时附带的cudatoolkit,所以必须保持两个cuda版本一致,否则冲突报错
CUDA安装教程
mmdet安装步骤
step1:pip install mmcv-full(根据对应torch版本和cudatoolkit版本)
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
step2:pip install mmdet
step3:pip install -r requirements/build.txt
pip install -v -e . # or "python setup.py develop"
step4: python mmdet_test.py
搜狗输入法
科研工具
-
Pycharm(编程IDE)
Pycharm Download 提取码: px33 -
mendeley(论文阅读和笔记)
Mendeley Download 提取码: 6ux8 -
Wiznote(记录软件)
Wiznote Download 提取码: eyaz -
TexStudio(Latex写作)
安装教程