在安装之前先看一下自己的显卡能不能安装CUDA，如果不能安装或者已经在英伟达官网下载CUDA并安装就跳到第5条。

在torch的官网有一键安装pytorch的命令，官网地址： https://pytorch.org/get-started/locally/

下面图片中为windows系统，如果有其他操作系统的请自行查看，为了避免不兼容，版本选择10.2。

安装torch和CUDA

只安装torch