CUDA error: invalid device ordinal

使用服务器进行单机多卡数据并行训练的时候，出现了一个小问题，本来以为可以直接创建4个process，可以4路数据并行，结果运行时发现报以下错：

RuntimeError: CUDA error: invalid device ordinal

原因：当前设备的GPU数量与cuda数量不匹配，
解决方案：
（1）先使用nvidia-smi命令查看服务器GPU数量

nvidia-smi

比如我这里看了一下，只有两张1080 Ti的卡

（2）根据GPU的数量改以下代码里的进程数量

（3）运行成功

#这里是调用cuda设备(GPU)
device = torch.device("cuda:{}".format(rank))

CUDA error: invalid device ordinal相关推荐

AssertionError: Invalid device id 和RuntimeError: CUDA error: invalid device ordinal
我在使用torch多卡并行时出现了这个两个问题. ##问题一:AssertionError: Invalid device id,即无效的设备id 出现的原因:结合代码解释: import ...os ...
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.根据 ...
cuda error invalid argument
用pytorch训练时报的错在loss.backward时报错估计应该是loss出现了异常值. 网上其他答案: 问题: RuntimeError: CUDA error: invalid conf ...
Cuda：invalid device pointer
cuda编程中调用一个判断cuda是否出错的函数如下: void check_gpu_error (const char *msg) {cudaError_t err = cudaGetLastErr ...
配置SSD-caffe测试时出现“Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal”解决
这是由于GPU数量不匹配造成的,如果训练自己的数据,那么我们只需要将solver.prototxt文件中的device_id项改为自己的GPU块数,一块就是0,两块就是1,以此类推. 但是SSD配置时 ...
CUDA error:invalid argument
当使用torch.view()时出现 RuntimeError: invalid argument 1: input is not contiguous at /pytorch/torch/lib/ ...
【CUDA开发】 Check failed: error == cudaSuccess (8 vs. 0) invalid device function
最近在复现R-CNN一系列的实验时,配置代码环境真是花费了不少时间.由于对MATLAB不熟悉,实验采用的都是github上rbg大神的Python版本.在配置Faster R-CNN时,编译没有问题, ...
显卡RTX 3090运行pytorch报错CUDA error: no kernel image is available for execution on the device
RuntimeError: CUDA error: no kernel image is available for execution on the device 安装适用于GeForce RTX ...
[debug] RuntimeError: CUDA error: no kernel image is available for execution on the device
问题描述运行程序时出现报错: RuntimeError: CUDA error: no kernel image is available for execution on the device ( ...

CUDA error: invalid device ordinal

CUDA error: invalid device ordinal相关推荐

最新文章

热门文章