Summary[问题简述（一句话）]
A clear and concise description of what the issue is.
Env & To Reproduce[复现与环境]
描述你用的环境、代码版本、模型
Screenshots[截图（如有）]
If applicable, add screenshots to help
Summary[问题简述（一句话）]
A clear and concise description of what the issue is.
Env & To Reproduce[复现与环境]
描述你用的环境、代码版本、模型
Screenshots[截图（如有）]
If applicable, add screenshots to help
1条答案
按热度按时间hrirmatl1#
检查一下你的pytorch版本，pytorch如果是cpu版本就改成cuda版本，其他issue里也有类似的问题，可以看一下和你遇到的问题符不符合。我也遇到了这个问题，网上搜是最新版的bug，我选择的办法是降级，用pytorch1.11配cuda11.3，希望能帮到你