AssertionError: Invalid CUDA ‘--device 1‘ requested, use ‘--device cpu‘ or pass valid CUDA device(s)

最新推荐文章于 2024-10-07 19:22:35 发布
最新推荐文章于 2024-10-07 19:22:35 发布
阅读量8k 收藏 8
点赞数 4
文章标签： 数学建模
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_49005550/article/details/130934341
版权

使用yolov5训练时，发下以下错误

错误提示：AssertionError: Invalid CUDA '--device 1' requested, use '--device cpu' or pass valid CUDA device(s)

解决方法：把下图这句代码修改为default='cuda:0'，

 结果如下：可以使用gpu了

 

lnad
关注
解决：AssertionError: CUDA unavailable, invalid device gpu requested
qq_42709514的博客
11-05 2万+
因为是刚接触机器学习，搞cpu的已经搞定，换gpu却出问题了，环境搞了一天。各种百度查。看到这里的朋友不要百度查了，直接上官网，上不了官网的参考一下方式：按照官网给的依赖包版本安装，不然容易踩依赖包版本不合适导致问题 PyTorch官网 CUDA 10.2 官网下载 CUDA 11.3 官网下载 CUDA 10.2 CUDA 10.2 conda ：conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch CUDA 1
解决pytorch报错:AssertionError: Invalid device id的问题
09-18
今天小编就为大家分享一篇解决pytorch报错:AssertionError: Invalid device id的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
RuntimeError: Invalid device string: ‘0‘
最新发布
h2313404476的博客
10-07 103
Using the option--gpu=cuda:0instead of--gpu=0 Or update the codetorch.cuda.set_device(args.gpu)withtorch.cuda.set_device(int(args.gpu))
yolov5环境搭建及遇到的问题Invalid CUDA ‘--device 0‘ requested, use ‘--device cpu‘ or pass valid CUDA device(）
hhhhhhh_hhhhhh_的博客
11-23 4107
2.配环境时一路顺风，但是最后运行detect.py只有CPU可以跑，选取device='0'的时候GPU总是报错，西安事变CUDA无效，原来是Pytorch版本的问题，（5）yolov5官方下载地址（release--->source版本）（4） 国内清华源镜像（配置国内源，加速下载）中由v1.8.2换为v1.9.0就可以了。（3）pytorch 对应cuda10.2。（1）miniconda下载。1.GPU主要参考某站。3.视频中用到的网址。
GPU训练Cuda报错
xiaoxian666的博客
10-09 6134
网站上查找对应自己cuda版本的pytorch。重新下载安装对应版本torchvision。卸载torchvision。
YOLOv8运行出错:ValueError: Invalid CUDA ‘device=1‘ requested. Use ‘device=cpu‘ or pass valid CUDA device
m0_47090348的博客
09-18 1万+
【代码】YOLOv8运行出错:ValueError: Invalid CUDA ‘device=1‘ requested. Use ‘device=cpu‘ or pass valid CUDA device。
AssertionError: Invalid CUDA ‘--device 0,1,2‘ requested, use ‘--device cpu‘ or pass valid CUDA devic
热门推荐
charlie_C9299的博客
08-01 2万+
如果不卸载cpuonly，运行下面的代码，依然会安装cpu版本的pytorch而不是gpu版本的pytorch。这里的cp38指的是python版本3.8.如果安装有问题，可能是whl文件的命名问题，可以用pip debug --verbose 命令来查看自己电脑支持的安装标签，然后修改whl文件的名字后在安装。这里面的cuda–version的版本是最高支持的cuda的版本，比这个低的是可以用的。在运行yolov5时，出现这个错误，意思是没有可用的cuda，无法使用GPU训练。是True也是不可用的。
解决Jetson Xavier NX上Invalid CUDA ‘--device 0‘ requested等问题
qq_33980935的博客
11-13 3745
所以下载“torch-1.8.0-cp36-cp36m-linux_aarch64.whl”以及“torchvision-0.9.0a0+01dfa8e-cp36-cp36m-linux_aarch64.whl”pwd=nyhx 提取码: nyhx。若不知道jetpack版本，可以通过jtop查看，我的版本是：jetpack 4.6 [L4T 32.6.1]主要原因：显卡驱动版本、CUDA版本、Pytorch版本不对应导致。说明安装的torch为CPU版本，卸载重新安装对应的GPU版本。
AssertionError: CUDA unavailable, invalid device 0 requested
weixin_49437670的博客
12-26 968
pip cuda和cudnn
CUDA_ERROR_INVALID_DEVICE
John Kang的博客
11-26 2348
TensorFlow 报错 failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_INVALID_DEVICE 解决方案：  问题1：回到主目录，重新运行代码。  问题2：  指定gpu时使用如下任意一种方式，而不是os.environ['CUDA_VISIBLE_DEVICES'] = "1" # 方法1 import os os.enviro...
AssertionError: Invalid CUDA --device 0’requested, use--device cpu’or pass valid CUDA device(s)
06-12
3. 如果您不需要使用CUDA，可以将代码中的`--device 0`参数改为`--device cpu`，以使用CPU运行代码。 4. 如果您需要使用CUDA，但是您的设备不支持CUDA，可以尝试使用云GPU等其他资源来运行您的代码。 希望这些解决...
AssertionError: Invalid CUDA '--device 1' requested, use '--device cpu' or pass valid CUDA device(s)
05-18
这个错误通常是因为你使用了一个无效的 CUDA ...你可以尝试使用 `--device cpu` 参数来使用 CPU 运行程序，或者使用正确的 CUDA 设备编号来指定可用的 CUDA 设备。例如，使用 `--device 0` 来使用第一个 CUDA 设备。
AssertionError: Invalid CUDA '--device 0' requested, use '--device cpu' or pass valid CUDA device(s)
05-10
2. 如果您的系统中没有 CUDA 设备或者您不想使用 CUDA，您可以使用 `--device cpu` 选项来让 PyTorch 使用 CPU 进行计算。 3. 确认您的 PyTorch 版本正确安装并且能够正常工作，您可以尝试使用以下命令来测试您的 ...
一些有关CUDA问题的解决总结
fmbao的博客
09-15 3074
最近在出差期间遇到了一些有关CUDA安装的问题，解决了好久，当然主要还是因为我菜的原因，把其中一些解决的路径记录下来以备后面不时之需。 删除原来的CUDA的方法 开始的时候我安装了一个CUDA10.1的版本，这是一个比较坑的版本，好多TensorFlow用不了，想着把原来的CUDA删除了，安装上新的CUDA版本，结果搞了半天，没整好。 后来找到了这个链接Ubuntu 18.04安装Tenso...
yolov5运行遇到的问题汇总
Conandoug的博客
12-06 5844
问题解决
关于报错AssertionError: CUDA unavailable, invalid device 0 requested的处理方式
pyxiaosheng的博客
03-29 1万+
从git上找了一个工程，按照其readme.md想运行试试，遇到报错： 这个报错是指：代码要运行CUDA，但是不能运行。 解决办法分两种情况：1.不运行CUDA；2.运行CUDA。 针对情况1： 点击蓝色 .\torch_utils.py 文件，传送到下图： 继续按照其readme.md运行试试，会遇到如下报错： 此时，点击报错中倒数第二个upsampling.py文件，传送到下图： 屏蔽后如下图： 到此，不运行CUDA的情况下可以正常运行程序了。 针对情...
评论 8
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值