文章目录
- 1、Pytorch配置
- 2、mmcv配置
1、Pytorch配置
- 查看虚拟环境+创建虚拟环境
conda env list：查看虚拟环境
conda create --name env_name python=3.7：创建虚拟环境
conda activate env_name：激活/进入该虚拟环境
- 查看自己的CUDA版本以及Python版本，根据这两者选择合适的torch和torchvision版本进行安装
查看CUDA版本可以使用nvidia-smi或者nvcc --version等命令进行查看
- 查看可以配置的torch和torchvision版本：
https://pytorch.org/get-started/previous-versions/
- https://download.pytorch.org/whl/torch_stable.html，进入该链接进行对应版本的torch和torchvision下载
- 最后直接pip install即可，例如：
pip install torch-1.12.1+cu113-cp37-cp37m-linux_x86_64.whl
2、mmcv配置
- 见链接：https://github.com/open-mmlab/mmcv/tree/1.x
pip install -U openmim
mim install mmcv-full==1.7.0
