torch gpu注意事项

记录一下torch使用gpu加速遇到的坑

1.版本对不上

https://github.com/pytorch/vision#installation
这个GitHub里列举了torch torchvision python三者应该对应的版本，我在安装对应版本上吃了大亏，来回装错了好几次才发现是版本问题。

2.安装gpu版本！！

看到torch.__version__返回值是1.10.2+cpu我就应该反应过来的，我直接用pip install torch下载的是cpu版本的torch，怎么对应版本都肯定是调用不了gpu的，我还恼羞成怒输了好几遍torch.version.cuda看看是不是电脑抽风了。
https://pytorch.org/get-started/locally/
去上面这个网址里查找你本地cuda版本对应的下载torch的命令，然后重新安装。

你本机的CUDA版本在cmd里输入
nvidia-smi
即可查看

我的CUDA是11.6，安装torch选11.3的即可。

3.安装成功后import torch还是报错

我的报错如下：

Key already registered with the same priority: GroupSpatialSoftmax

这个是因为反复安装torch导致的问题，一般来说彻底删除torch可以解决，尝试了一下这个帖子
帖子
发现对我来说没啥用，干脆重新创建了虚拟环境，完美解决。

目前就这么几个大坑，以后再碰到了再继续补。