文章目录
- 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文章来源:https://uudwc.com/A/qREbR
2、mmcv配置
- 见链接:https://github.com/open-mmlab/mmcv/tree/1.x
pip install -U openmim
mim install mmcv-full==1.7.0文章来源地址https://uudwc.com/A/qREbR