Pytorch使用GPU加速的步骤

---

CUDA（Compute Unified Device Architecture），是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。近些年来，显卡的计算能力越来越强大，如果只是进行图形计算未免有些浪费性能，这个并行运算架构应运而生。首先，使用GPU加速前，确认自己硬件是否支持CUDA，可以自己去百度自己的显卡类型。可以在cmd窗口输入nvidia-smi命令来确认自己的显卡是否有CUDA。如果该命令无效，建议先去更新自己的显卡驱动，或者用360驱动大师检查驱动和更新。

看第一行右边，如果没有显示CUDA版本，则需要自己安装CUDA。

然后安装CUDA。网址：https://developer.nvidia.com/zh-cn/cuda-downloads，可以自行下载安装包并安装。安装好后，再次输入nvidia-smi命令检查版本。

然后安装Pytorch，必须安装的有torch和torchvision两个包，注意不要直接在Pycharm中自动下载安装ÿ