pycharm报错:AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘
·
原因:pytorch没有配置GPU/CUDA(若已经安装并配置GPU+CUDA)
到pytorch官网安装适合的版本:
https://pytorch.org/get-started/locally/

将最底层的 Run this Command复制,到pycharm的终端运行。
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐



所有评论(0)