Pytorch"填坑“记（长更）

最新推荐文章于 2021-08-12 09:56:21 发布
最新推荐文章于 2021-08-12 09:56:21 发布
阅读量523 收藏
点赞数
分类专栏： pytorch 文章标签： pytorch GPU
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/lihangll/article/details/89112355
版权

问题：

存在多个ＧＰＵ时，指定GPU时的错误：RuntimeError: cuda runtime error (10) : invalid device ordinal at /pytorch/torch/csrc/cuda/Module.cpp:34


解决方法之一：

CUDA_VISIBLE_DEVICES=6（根据自己的ＧＰＵ个数，任意改变）
./train.py --gpu 0(不要改变它！！！从０开始）

原因之一：

因PyTorch程序设置的gpu_id编号未从0开始导致

Hang-Hang Li
关注
专栏目录
PyTorch~cpp_extension
whaosoft143ai的博客
12-19 533
还是pytorch哈~~ 结合 Python 与 C++ 各自的优点，在 PyTorch 中加入 C++ / CUDA的扩展，详细解释C++/CUDA 算子实现和调用全流程本期主题：结合 Python 与 C++ 各自的优点，在 PyTorch 中加入 C++ / CUDA的扩展，而让我们自己更好地使用工具而不为工具所束缚。代码来源：MMCV, PyTorch。注：C++ / CUDA 扩展一般有”预编译“ 与 ”实时编译“ （just-in-time, JIT）模式。本期主要介绍”预编译“模式。
PyTorch 1.0 中文文档：torch.utils.cpp_extension
weixin_30469895的博客
03-07 1972
译者: belonHan torch.utils.cpp_extension.CppExtension(name, sources, *args, **kwargs) 创建一个C++的setuptools.Extension。 便捷地创建一个setuptools.Extension具有最小（但通常是足够）的参数来构建C++扩展的方法。 所有参数都被转发给setuptools.Extension...
pytorch拓展-cpp extension
CV老李的博客
06-09 1391
pytorch的cpp拓展
pytorch中为Module和Tensor指定GPU
兔角与禅
06-02 7606
pytorch指定GPU 在用pytorch写CNN的时候，发现一运行程序就卡住，然后cpu占用率100%，nvidia-smi 查看显卡发现并没有使用GPU。所以考虑将模型和输入数据及标签指定到gpu上。 pytorch中的Tensor和Module可以指定gpu运行，并且可以指定在哪一块gpu上运行，方法非常简单，就是直接调用Tensor类和Module类中的 .cuda() 方法。 i...
RuntimeError: CUDA error (10): invalid device ordinal
博客
11-30 9060
在运行cuda程序时发生报错，出现 RuntimeError: CUDA error (10): invalid device ordinal   造成这个错误的原因主要是本地只有一个 GPU (GPU:0)，而程序中使用 GPUs:1。 在配置信息中，程序是1，我改成0就可以正常运行了。  ...
Pytorch版Faster R-CNN配置运行填坑记录
Watermelon1986的博客
11-28 924
Pytorch Faster R-CNN配置运行填坑记录 之前一直在用Nvidia GTX1080Ti做实验，用到的代码及整套开发环境的版本都较老（Python2.7+Pytorch0.4.0+CUDA8.0），由于20系列显卡或30系列显卡都不再支持CUDA8.0，所以需要进行一次整体升级，在此做一个升级配置环境的填坑记录。
Pytorch的坑
qq_41879767的博客
03-20 408
关于Anaconda环境下的Pytorch安装的TipsAnaconda的安装怎么安装AnacondaPytorch的安装让我们一起来填坑吧 Anaconda的安装 整个戏剧故事的开始自然是要从Anaconda开始了 不过我在安装pytorch的时候其实老早就已经安装好anaconda的环境了 但是作为故事发展需要还是有必要提一下这一点的 ...
win10中pytorch以及torchvision安装填坑
smilexuxu123的博客
07-04 1101
安装pytorch以及torchvision过程中遇到了一些问题，记录下来以便日后复习，也希望给遇到相同问题的小伙伴一些启发。 1.安装pytorch anaconda和cuda的安装过程这里不再说明，本文使用的是python3.6和cuda8.0 参考网上一些大佬的教程用conda命令执行官网上的相关命令时，安装过程实在是太慢了，使用了清华源也不能很好地得到解决。于是转战pip3；使用py...
学习pytorch的小记录
qq_29320285的博客
03-12 213
学习pytorch的小记录原教程地址1.PyTorch神经网络简介1.1 计算梯度1.2 这一部分的可运行代码贴在下面2.训练一个分类器2.1 如何进行数据处理2.2 训练的步骤2.3 数据处理目前暂时后面的还没看完~ 原教程地址 PyTorch简明教程 1.PyTorch神经网络简介 1.1 计算梯度 那里是有一点不太明白的记录一下。 在调用loss.backward()之前，我们需要清除掉te...
PaddleOCR填坑记
Dorthyn的博客
08-12 1203
PaddleOCR部署
pytorch 使用指定的GPU RuntimeError: CUDA error: invalid device ordinal
Hello Word!
07-31 3万+
pytorch使用指定GPU报错： Traceback (most recent call last): File "test_bed/process_deepglint.py", line 102, in <module> pred_dataset(outputFile) File "test_bed/process_deepglint.py", line 36, in pred_dataset pred_loader_deepg, model, criterion,
Python 错误 RuntimeError: CUDA error (10): invalid device ordinal
热门推荐
linchunmian的博客
05-31 5万+
一、Python Error在 Pytorch 读取参数时，报错 RuntimeError: cuda runtime error (10) : invalid device ordinal。二、解决方法造成这个错误的原因主要是本地只有一个 GPU (GPU:0)，而程序中使用 GPUs:1。因此，在程序中找到定义 device 的地方，我的如下：将其中 "cuda:1" 修改为 "cuda:0"...
PyTorch Error fix-RuntimeError: kind_.is_prim() INTERNAL ASSERT FAILED
夏普通
07-26 1706
报错如下： RuntimeError: kind_.is_prim() INTERNAL ASSERT FAILED at /pytorch/torch/csrc/jit/ir.cpp:904, please report a bug to PyTorch. Only prim ops are allowed to not have a registered operator but aten::_convolution doesn't have one either. We don't know if t
pytorch模型提示超出内存cuda runtime error(2): out of memory
学习、记录、分享
11-23 2万+
看到这个提示，表示您的GPU内存不足。由于我们经常在PyTorch中处理大量数据，因此很小的错误可能会迅速导致程序耗尽所有GPU; 好的事，这些情况下的修复通常很简单。这里有几个常见检查事项包括： 一、不要在循环训练中累积历史记录。 默认情况下，涉及需要求导/梯度gradients变量的计算将保存在内存中。计算中避免使用这些变量，例如在跟踪统计数据时，这些变量在循环训练中将超出你内存。相反，您...
pytorch 的 CUDA 编程 CUDAExtension
lianbus的专栏
06-28 4435
通常我们在使用pytorch设计和训练网络后，需要对网络输出的结果再做一定后处理，才能得到我们的想要的结果。 通常的方法是将网络输出的结果，复制回CPU，并以numpy形式进行数据的后处理，但是对于任务很重的后处理，比如对场数据或者图像数据，做这些操作就会耗费大量的时间，更理想的做法是使用pytorch提供的CUDA扩展，直接在GPU中处理好这些数据，只将最终的结果返回给CPU，一方面并行计算可...
Pytorch使用Python,CPP,CUDA进行拓展
loopun的博客
07-04 1385
使用CPP 和CUDA进行扩展时面临着几个小问题：python与cpp的接口绑定方式， python与cpp的数据交互方式，cpp编译工具的选择。现在逐条回答： 数据接口 ATen: The foundational tensor and mathematical operation library on which all else is built; ATen ATen is fundame...
解决RuntimeError: cuda runtime error (30) : unknown error at /pytorch/aten/src/THC/THCGeneral.cpp:70&N...
ZHO9504的博客
11-14 2万+
解决两个问题： （1）RuntimeError: cuda runtime error (30) : unknown error at /pytorch/aten/src/THC/THCGeneral.cpp:70 （2）NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make...
使用pip安装PyTorch时间太长
最新发布
05-14
如果使用pip安装PyTorch时间较长，可能是因为网络不稳定或者下载速度较慢。你可以尝试以下方法： 1. 使用清华大学的镜像源进行安装，可以通过以下命令进行设置： ```python pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch ``` 2. 使用conda进行安装，conda是一个比pip更快、更稳定的包管理工具，你可以先安装Anaconda或Miniconda，然后使用以下命令安装PyTorch： ```python conda install pytorch torchvision torchaudio -c pytorch ``` 3. 如果你已经安装了PyTorch，可以尝试使用以下命令更新： ```python pip install --upgrade torch torchvision torchaudio ``` 4. 如果以上方法都无法解决问题，可以尝试使用代理或者VPN进行安装。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值