![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
文章平均质量分 75
melicent114
这个作者很懒,什么都没留下…
展开
-
python+cuda编程(二)
请注意,上面核函数的参数列表是void,而我们在Python中调用该函数时却给予了一个参数block,该参数的格式必须是一个长度为3元组,而且元组元素的类型为int。__global__是一个CUDA的关键字,它出现在一个函数的前面,限定这是一个CUDA核函数(kernel function),而不是C++中的函数。我想你一定发现了,核函数的执行次数就是里面的数字的乘积。如果你的CUDA环境没有问题,那么下面内容将会输出在你的屏幕。找到对应的版本即可,这里可以不下载,用pip命令下载安装。原创 2022-09-28 19:06:00 · 1417 阅读 · 0 评论 -
python+cuda编程(一)
numbaNumba通过及时编译机制(JIT)优化Python代码,Numba可以针对本机的硬件环境进行优化,同时支持CPU和GPU的优化,并且可以和Numpy集成,使Python代码可以在GPU上运行,只需在函数上方加上相关的指令标记,PyCUDAPyCUDA的内核函数(kernel)其实就是使用C/C++编写的,通过动态编译为GPU微码,Python代码与GPU代码进行交互。对比。原创 2022-09-28 18:53:14 · 5811 阅读 · 1 评论 -
ubuntu安装mmdetection
ubuntu安装mmdetection原创 2022-07-06 09:41:46 · 980 阅读 · 2 评论 -
转权重格式darknet转onnx转tensorrt
将darknet格式的权重转为onnx格式,然后再转为tensorrt格式首先准备好一个darknet格式的权重文件(.weights),配置文件(.cfg),训练集名称文件(.names)darknet转为onnx这里我使用的是这个开源链接https://github.com/Tianxiaomo/pytorch-YOLOv4图1通过一个python脚本,将darknet格式的权重,转为onnx格式,并且可以输入一张图片,这个脚本会对这张图片进行识别具体的步骤为:(1)先把上面这个gi原创 2022-03-20 23:34:42 · 1606 阅读 · 1 评论 -
ubuntu安装tensorrt
ubuntu安装tensorrt前提:先安装好cuda和cudnn步骤一:下载tensorrt压缩包。找到自己对应cuda版本的安装包从nvidia的官网下载https://developer.nvidia.com/nvidia-tensorrt-download(由于我安装的是cuda11,所以我这里下载的是下面这个)步骤二:下载下来的tensorrt压缩包为TensorRT-8.0.1.6.Linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz,这里放置在服务原创 2022-03-20 23:12:35 · 5107 阅读 · 2 评论