Win10系统下安装PyTorch
最近由于重装了系统,所以需要在机器上重新建立PyTorch环境。在网上查了一些资料,发现很多都已经过时了。以前的比较方便的一种安装流程为:
- Anaconda安装Python,这种方式比较简单且有效;
- 对于使用GPU的情况下,安装CUDA和cuDNN;
- 到Pytorch官网安装相应版本的PyTorch。
然而目前PyTorch在安装中已经包含了相应的工具,所以当前安装过程已经大大简化。与上面类似,可以直接使用下面流程:
- Anaconda安装Python;
- 通过一条命令建立使用GPU的PyTorch:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
不同的硬件平台可以在https://pytorch.org/选择相应合适的命令。
安装cv2包
在安装cv2包的时候,发现网上很多教程都是通过下载源码包进行安装,比较复杂。我们这里直接考虑使用pip来进行安装:
pip3 install opencv-python
网友评论