Win10 设置Pytorch(GPU版）

最近开始做视频行为检测的相关工作，第一步当然是设置环境了，在这里记录一下相关过程。

第一步：安装Anaconda

Anaconda之前电脑上就已经安装过啦，这里就不详细说了，如果有需要的朋友可以去Anaconda的官网上进行下载，安装即可。

第二步：安装CUDA ToolKit, cuDNN

• 2.1 CUDA版本确定

        简单来说，显卡 = GPU，我的电脑使用的是NVDIA公司生产的显卡，型号为 RTX 2070。如果有朋友需要确定自己的显卡，可以通过：控制面板 --> 设备管理器 --> 显示适配器，进行确认。

       CUDA 是NVDIA推出的用于自家GPU的并行计算框架。注意，并不是所有GPU都支持CUDA。只能在NVIDIA的GPU上运行CUDA。CUDA版本可以cmd中使用nvidia-smi命令查看：通过查看，我的CUDA版本为11.2