PyTorch安装使用

为啥要用PyTorch

人工智能不跟风，那就没有进步，没有发展。

PyTorch很火，不跟不行，必须得跟，得用、

如何使用PyTorch

PyTorch使用的前置条件

PyTorch用的好，前置条件少不了。

初学的咱用Windows系统就很好，Anaconda， python 3.x 两个基石必须有。搞深度学习，没个显卡咋弄。GPU搞起来，那就得有CUDA来支持。
1、Anaconda3里有python3.x了，所以装好Anaconda 3好了，官网下载安装Anaconda，一气呵成，Windows下双击单击666.
2、CUDA版本搞清楚。
关于CUDA与在Anaconda 下安装PyTorch的cudatoolkit区别，参考
https://blog.csdn.net/xiqi4145/article/details/110254093

安装PyTorch

来到PyTorch官网， https://pytorch.org/get-started/locally/，根据自己硬件情况选择相应的选项。
至于选择哪个CUDA版本，右击电脑桌面，NVIDIA控制面板–>帮助–>系统信息–>组件–>3D设置就可以看到CUDA版本号了。

例如：

pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

一般用pip安装成功概率大些，一次不成多试几次，总能成功。

PyTorch用起来

要想看看安装是否成功，来个代码试一下

# -*- coding: utf-8 -*-
# @Time    : 2021/5/17 16:59
# @Author  : HelloWorld！
# @FileName: pytorchtest.py
# @Software: PyCharm
# @Operating System: Windows 10
# @Python.version: 3.6

import torch
import torchvision
print(torch.__version__)
x = torch.rand(5, 3)
print(x)
print(torch.cuda.is_available())

如果如下所示，则运行成功

Python 3.8.10 (default, May 19 2021, 13:12:57) [MSC v.1916 64 bit (AMD64)] on win32
runfile('D:/AIProject/FallDetect/pytorchtest.py', wdir='D:/AIProject/FallDetect')
1.8.1+cu111
tensor([[0.3789, 0.4321, 0.3657],
        [0.7428, 0.2408, 0.8792],
        [0.5774, 0.4373, 0.0717],
        [0.6202, 0.6959, 0.7289],
        [0.3364, 0.4481, 0.2087]])
True

在后续的使用中，需要安装依赖包，包括opencv等
例如：

pip install opencv-contrib-python
pip install pandas
pip install matplotlib