使用anaconda配置pytorch及openCV

这里我们默认电脑已安装并配置好了anaconda

一、创建虚拟环境

首先在开始菜单栏找到并打开anaconda prompt

创建一个新的虚拟环境，在这里我将其命名为pytorch,python版本使用3.6

版本对应关系如下图所示，也可参照GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision

最新torch版本参见第二部分的torch官网

 

conda create -n pytorch python=3.6

参看当前存在的环境列表

conda env list

进入pytorch环境

conda activate pytorch

此时的（base）应变为（pytorch），下面我们先安装pytorch

二、配置pytorch

首先进入pytorch官网
PyTorchhttps://pytorch.org/根据自己电脑的情况选择不同选项

如果电脑装有英伟达显卡，可以通过输入nvidia-smi参看CUDA版本号

 复制最下面一行的命令，等待分析完毕后输入y，即可自动安装

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

在这里，由于使用默认源下载速度过慢，建议使用清华源或其他源进行下载

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
# reference
# https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

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

如使用其他源，去掉末尾的-c pytorch

conda install pytorch torchvision torchaudio cudatoolkit=11.3

参看当前package列表

pip list

可以看到已存在torch 

最后，经检验，pytorch已可以成功导入并使用。

三、配置openCV

直接安装即可

pip install opencv-python

参看package列表

pip list

可以看到已存在opencv-python 

 下面提供一段简短的测试代码

import torch
import cv2

src = cv2.imread('sky1.jpg', cv2.IMREAD_COLOR)
cv2.imshow("img", src)
cv2.waitKey()

a = torch.ones(3, 3)
print(a)