使用torch的一定注意cuda版本必须要跟torch版本对应，网上好多网友写的博客都没提到，目前cuda官网的版本是11.0但是我当时在torch官网下载的是支持cuda10.2版本的torch。
cuda 10.2 链接是cuda10.2版本的下载链接，下载之后按照提示步骤一步步安装即可非常简单。
两种方法测试是否安装成功：
1.进入cmd,输入nvcc -V，会出现cuda的版本号等相关信息。
2.如果正在用torch的话，直接输torch.cuda.is_available()，如果返回True则表示cuda安装成功，我们可以用GPU训练神经网络。