AssertionError: CUDA unavailable, invalid device 0 requested

在使用yolov5训练时，提示这个错误，查看torch.cuda.is_available()得到的返回值是false

查看nvidia-smi可以看到信息

由于之前使用conda安装的torch1.9.0

这里使用pip重新安装：

pip3 install torch == 1.9.0+cu111 torchvision == 0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

全部安装后，再次执行torch.cuda.is_available()时

可以看到正确结果了

参考：AssertionError: CUDA unavailable, invalid device 0 requested_消灭BUG鸭的博客-CSDN博客_cuda unavailable

AssertionError: CUDA unavailable, invalid device 0 requested相关推荐

AssertionError: Invalid device id 和RuntimeError: CUDA error: invalid device ordinal
我在使用torch多卡并行时出现了这个两个问题. ##问题一:AssertionError: Invalid device id,即无效的设备id 出现的原因:结合代码解释: import ...os ...
Cuda：invalid device pointer
cuda编程中调用一个判断cuda是否出错的函数如下: void check_gpu_error (const char *msg) {cudaError_t err = cudaGetLastErr ...
【CUDA开发】 Check failed: error == cudaSuccess (8 vs. 0) invalid device function
最近在复现R-CNN一系列的实验时,配置代码环境真是花费了不少时间.由于对MATLAB不熟悉,实验采用的都是github上rbg大神的Python版本.在配置Faster R-CNN时,编译没有问题, ...
pytorch测试报错：RuntimeError: cuda runtime error (10) : invalid device ordinal at torch/csrc/cuda/Module
模型在服务器多gpu上训练,测试在自己台式机上进行,只有一块gpu,测试报错: File "/home/fuxueping/sdb/PycharmProjects/face_recognit ...
Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal
ubuntu14.04 caffe 运行别人的程序,报错Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal.根据 ...
RuntimeError: Expected object of device type cuda but got device type cpu for argument pytorch数据位置
RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'target' i ...
Caffe + Ubuntu 15.04/16.04 + CUDA 7.5/8.0 在服务器上安装配置及卸载重新安装（已测试可执行）
本文参考如下: caffe 安装所需的所有资源可在百度网盘下载链接: http://pan.baidu.com/s/1jIRJ6mU 提取密码:xehi 在服务器上为每个子用户拷贝caffe 使用 ...
nvidia cuda工具包更新9.0版本记录
因工作需要,对公司2台服务器升级cuda到9.0版本.本文记录其过程. 介绍 cuda在nvidia官方网站有下载,最新版本为9.2,但公司采用的tensorflow只使用9.0,为了版本的兼容,决定 ...
DPDK报错：Ethdev port_id=0 requested Rx offloads 0xe doesn‘t match Rx offloads capabilities 0x800
运行OpenNetVM的NF Manager时报错如下:Ethdev port_id=0 requested Rx offloads 0xe doesn't match Rx offloads cap ...
常识——windows的tensorflow安装gpu版本,cuda算力只有3.0也可以
文章目录问题 windows安装tensorflow_gpu版本检查显卡是否支持cuda 查看自己的显卡型号去官网查看是否支持cuda 检查显卡的cuda算力按照GPU算力3.0的电脑配置进行 ...

AssertionError: CUDA unavailable, invalid device 0 requested

AssertionError: CUDA unavailable, invalid device 0 requested相关推荐

最新文章

热门文章