在AMD GPU上nvidia-smi的等价指令

背景介绍我们在使用Pytorch训练时，模型和数据有可能加载在不同的设备上（gpu和cpu），在算梯度或者loss的时候，报错信息类似如下：RuntimeError: Function AddBackward0 returned an invalid gradient at index 1 - expected type torch.cuda.FloatTensor but got torch.F

题外话：经常遇到这个问题，每次都google，还是记录一下好了 >.<通常，在一个新的系统环境下，跑CNN或者DL的python代码，会报错找不到cv2 module。环境里没有安装opencv导致的，问题错误通常如下：ModuleNotFoundError: No module named 'cv2'通过pip指令安装opencv-python包即可：pi...

题外话：这次是被编译顺序坑了很久。。。还是基础学的不扎实实验背景：用g++编译cpp文件，依赖于opencv。待编译的cpp文件cv_test.cpp内容如下：#include "opencv2/core/mat.hpp"int main(){cv::Mat mat;return 0;}该文件仅用于测试opencv链接库是否编译...

最大流/最小割（Max-Flow/Min-Cut）算法，因其可以对能量方程最小化进行求解，在计算机视觉（CV）中有很广泛的应用。之前介绍过的Seam Carving算法（用于图像智能缩放），或是图像分割算法中，经常使用最大流/最小割进行能量方程的最小化求解。

题外话：可能因为是节假日，所以这篇文章的标题非常的标题党哈哈哈哈哈~ 回归严肃，过去一段时间花了比较多的精力学习这个算法，所以想借此空闲时间，好好总结和复盘一下。感兴趣的朋友们，欢迎评论区讨论，如果有理解不对的地方也欢迎大家指正。前言车道线检测在自动驾驶感知任务中占据重要的一席，它可以辅助自动驾驶的车道保持，同时也可以为车辆定位等工作提供较强的先验信息。目前，常见的方法大抵...

目的：使用tensorboard可视化Pytorch模型进展：Pytorch1.1.0版本，并使用pip install tensorboard报错信息：No module named 'torch.utils.tensorboard' 提示找不到tensorboard搜到的原解决方案：https://www.endtoend.ai/blog/pytorch-tensorb...

在使用GPU进行AI模型训练或推理部署的时候，我们通常需要确认当前系统中的硬件信息，如GPU型号、GPU数量、GPU可用显存等等。在NVIDIA GPU上，指令nvidia-smi默认打印以上所有的常用信息。对应来说，AMD GPU中常用指令rocm-smi，打印信息中目前（2024年12月24日平安夜）还不包含GPU型号。待解决问题：如果希望查看当前系统中所配置的AMD GPU的型号，应该如何操

背景介绍我们在使用Pytorch训练时，模型和数据有可能加载在不同的设备上（gpu和cpu），在算梯度或者loss的时候，报错信息类似如下：RuntimeError: Function AddBackward0 returned an invalid gradient at index 1 - expected type torch.cuda.FloatTensor but got torch.F

背景在nvidia ngc平台下载docker image，根据实验要求在环境里安装opencv。因为当前实验tensorflow要求版本较低，我下载了比较久远的版本镜像。如果大家是自己复现代码，还是使用尽可能新的环境，则可以避免大部分环境问题。问题通过pip安装opencv，指令如下：pip install opencv-python安装过程的输出：Collecting opencv-pytho