在安装之前先看一下自己的显卡能不能安装CUDA,如果不能安装或者已经在英伟达官网下载CUDA并安装就跳到第5条。
在torch的官网有一键安装pytorch的命令,官网地址: https://pytorch.org/get-started/locally/
下面图片中为windows系统,如果有其他操作系统的请自行查看,为了避免不兼容,版本选择10.2。
安装torch和CUDA
只安装torch