栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Pytorch安装及GroupSpatialSoftmax报错解决

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/822194.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号