Win10系统下安装PyTorch、cv2及其它相关包-2022

Win10系统下安装PyTorch

最近由于重装了系统,所以需要在机器上重新建立PyTorch环境。在网上查了一些资料,发现很多都已经过时了。以前的比较方便的一种安装流程为:


  1. Anaconda安装Python,这种方式比较简单且有效;
  2. 对于使用GPU的情况下,安装CUDA和cuDNN;
  3. 到Pytorch官网安装相应版本的PyTorch。

然而目前PyTorch在安装中已经包含了相应的工具,所以当前安装过程已经大大简化。与上面类似,可以直接使用下面流程:


  1. Anaconda安装Python;
  2. 通过一条命令建立使用GPU的PyTorch:
    conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

不同的硬件平台可以在https://pytorch.org/选择相应合适的命令。

安装cv2包

在安装cv2包的时候,发现网上很多教程都是通过下载源码包进行安装,比较复杂。我们这里直接考虑使用pip来进行安装:
pip3 install opencv-python

你可能感兴趣的:(Win10系统下安装PyTorch、cv2及其它相关包-2022)