centos7安装pytorch

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

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

1 conda create -n open-mmlab python=3.7 -y
2 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版本查看:

1 import torch
2 print(torch.__version__)
原文地址:https://www.cnblogs.com/haoyul/p/12212445.html