Pytorch安装及GroupSpatialSoftmax报错解决

作者: pdnbplus | 发布时间: 2024/07/14 | 阅读量: 243

# 项目场景:

安装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,也能照常使用