【Pycharm/Anaconda配置环境记录】

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

猜你喜欢

转载自blog.csdn.net/m0_48086806/article/details/133279093