Pytorch安装及GroupSpatialSoftmax报错解决

项目场景： 安装Pytorch，安装中出现的问题

在官网下载pytorch

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

第一个BUG：Key already registered with the same priority: GroupSpatialSoftmax

安装完进入python,输入以下代码

import torch

会报如下错误Key already registered with the same priority: GroupSpatialSoftmax

解决方案：

卸载两次pytorch，再卸载一次，再重装(不会真正的重装，会有缓存)

pip uninstall torch
pip uninstall torch
conda uninstall pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

再次进入python运行以下代码

import torch
torch.cuda.is_available()
>>> True

第二个BUG：Failed to start the Kernel

这是用vscode打开装了ipykernel的jupyter notebook的时候报的错

解决方案 重装pyzmq

pip uninstall pyzmq # 卸载pyzmq
pip install pyzmq==19.0.2 #安装pyzmq19.0.2版本

Pytorch

装pytorch不需要另外去匹配CUDA的版本与Python的版本，只有机子里面有CUDA一般都会有，就不用另外装，cudatoolkit这玩意儿估计能解决版本的问题,我用的就是python3.6然后环境里面是9.0的CUDA，也能照常使用