pytorch安装（GPU）

真的不要轻易同时安装两个版本的包，看到网上说已经有了cpu版本的pytorch，还是可以直接安装下载gpu版本的，结果导入torch就报错了（Key already registered with the same priority: GroupSpatialSoftmax），没办法，只能两次uninstall，再重新安装。

安装pytorch的过程就是不断踩坑不断踩坑的过程，真的是，不知道自己想的，本来好好的pytorch不用，非要去尝试使用gpu。（然后GPU跑代码，嗯，真香哈哈哈哈哈哈哈哈哈哈）

1.首先在命令行输入nvidia-smi，来查看驱动版本和CUDA版本

2.在NVIDIA官网，看cuda版本以及GPU驱动的对应关系
3.然后去Pytorch官网，根据Compute Platform选择相应的pytorch进行安装。

4.就按照官网的命令来运行，安装成功，但是import又报错。

5.最后实在太懒，就直接新建环境（我这里命名为pytorch，前面都是base），在环境里面安装下载了1.10.0GPU版本的pytorch，并且使用torch.cuda.is_available()命令，结果为True，证明GPU是可以正常使用的。

关于环境管理的part，我之前有写过一篇博文专门介绍，包括版本更新后，博文里面的命令也修改过，可以点击这里，进行查看。