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

【PyTorch】Key already registered with the same priority: GroupSpatialSoftmax 解决方法

Python 更新时间:发布时间: 百科书网 趣学号

Key already registered with the same priority: GroupSpatialSoftmax
  • 问题描述
  • 原因分析
  • 解决方法

问题描述

记得之前安装PyTorch的GPU版本安装成功了 PyTorch 最新安装教程(2021-07-27)

可最近使用GPU时,发现为False不可用!

import torch
torch.cuda.is_available()
原因分析

可能是最近在PyCharm中安装torchtext模块时,附带安装PyTorch-CPU版本,因此覆盖掉原来的GPU版本,两个版本的torch folder名字重合了,conda安装和pycharm安装包其在运行程序的时优先级上起了冲突,因此报错!

解决方法

进入Anaconda中使用的环境:

activate virtualEnvs

1.将两个版本的torch全部卸载,否则可以成功import torch,但实际上无法运行。

# 需要跑两次pip uninstall
pip uninstall torch 

2.重新安装所需要的版本

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

参考:link


加油!

感谢!

努力!

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

版权所有 ©2023-2025 051e.com

ICP备案号:京ICP备12030808号