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

AssertionError: Torch not compiled with CUDA enabled

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

AssertionError: Torch not compiled with CUDA enabled

简单解决方法:

在代码前添加如下代码:

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

  再将代码其余地方出现.cuda()的地方改成.to(device),就可以在无gpu的环境中运行了。

例如:

 参考:(57条消息) (树莓派)解决问题:AssertionError: Torch not compiled with CUDA enabled_Clark-dj的博客-CSDN博客https://blog.csdn.net/dujuancao11/article/details/114006234

不简单的:

   代码本身可以确定是没有问题的,报错含义是在编译torch的时候,CUDA并没有起作用。

检查Anaconda prompt 的基本环境中,是否装有Pytorch(GPU版)。

经检查,在Anaconda的基本环境中没有安装Pytorch(GPU版),所以Pycharm即使引入了conda environment,也没有办法使用Pytorch(GPU版)。

解决:安装Pytorch(GPU版)

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

 还是失败了 T-T。

参考:(57条消息) 解决AssertionError Torch not compiled with CUDA enabled问题_这也太南了趴的博客-CSDN博客https://blog.csdn.net/qq_46941656/article/details/119701547?spm=1001.2101.3001.6650.5&utm_medium=distribute.pc_relevant.none-task-blog-2~default~BlogCommendFromBaidu~Rate-5-119701547-blog-124561850.pc_relevant_paycolumn_v3&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2~default~BlogCommendFromBaidu~Rate-5-119701547-blog-124561850.pc_relevant_paycolumn_v3&utm_relevant_index=8

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

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

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