win10安裝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

 pytorch官網的輸入是：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

 

若遇到以下錯誤：

 

 

 

 

 

 

 

 

 

 

解決方法：

　　a.更改鏡像：因為官網的速度實在太慢太慢了

添加清華鏡像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ 
conda config --set show_channel_urls yes

添加pytorch鏡像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

　　b.安裝pytorch：從官網獲得conda指令 :conda install pytorch torchvision cudatoolkit=9.0   因為修改了鏡像，所以去掉了官網的給的地址（-c pytorch）

我的輸入命令：

conda install pytorch torchvision cudatoolkit=10.1

　　c.驗證是否安裝成功：

import torch
print(torch.__version__)

 

 

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__)