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

解决:AssertionError: CUDA unavailable, invalid device gpu requested

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

解决:AssertionError: CUDA unavailable, invalid device gpu requested

因为是刚接触机器学习,搞cpu的已经搞定,换gpu却出问题了,环境搞了一天。各种百度查。看到这里的朋友不要百度查了,直接上官网,上不了官网的参考一下方式:按照官网给的依赖包版本安装,不然容易踩依赖包版本不合适导致问题 PyTorch官网

CUDA 10.2 官网下载
CUDA 11.3 官网下载

PyTorch官网 给出的 CUDA 10.2 安装依赖


CUDA 10.2 conda :conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
CUDA 10.2 pip :pip3 install torch1.10.0+cu102 torchvision0.11.1+cu102 torchaudio===0.10.0+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

PyTorch官网 给出的 CUDA 11.3 安装依赖


CUDA 11.3 conda :conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
CUDA 11.3 pip :pip3 install torch1.10.0+cu113 torchvision0.11.1+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

注:我遇到了这样一个问题,最后可以运行了不知为啥。环境是yolov5, anaconda虚拟环境 python 3.8.8

单独运行torch是可以的

运行yolov5的 detect.py 却报错


最可笑的是在运行整个识别之前导入torch之后马上调用一下:torch.cuda.is_available() 竟然可以了:表示不懂为什么有大佬懂得给指点一下

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

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

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