centos7安装pytorch

　　pytorch地址：https://pytorch.org/get-started/locally/

 

1.创建一个从打虚拟环境并激活它

conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab

 

2.安装官方说明安装pytorch和torchvision，例如：

conda install pytorch torchvision -c pytorch

 我的输入是：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

 

 

3.克隆mmdetection存储库

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection

 

4.安装mmdetection（其他依赖库将自动安装）

pip install mmcv
python setup.py develop  # or "pip install -v -e ."

 

注意：

1. git commit id将在步骤d中写入版本号，例如0.6.0 + 2e7045c。该版本还将保存在经过训练的模型中。建议您每次从github提取一些更新时都运行步骤d。如果修改了C ++ / CUDA代码，则此步骤为强制性的。
2. 按照上述说明，mmdetection将安装在dev模式下，对代码进行的任何本地修改都将生效，而无需重新安装它（除非您提交了一些提交并希望更新版本号）。
3. 如果要使用opencv-python-headless而不是opencv-python，可以在安装MMCV之前先安装它。

 

 

 

2.验证安装是否成功

进入python，输入命令：

# python

pytorch版本查看：

import torch
print(torch.__version__)