- 博客(5)
- 收藏
- 关注
RSS订阅原创 理解深度学习中的自动微分
所有满足条件的tensor梯度会自动保存到对应的grad属性里，而你需要求梯度的叶子结点一定是满足上述条件的，而计算过程中的非叶子节点的 .grad 属性默认情况下是不会被保存的，因为它们通常不是计算梯度的终点（这是torch框架的考虑设定）。因为指定了输出为y，输入为x，所以返回值就是 y关于x的梯度，完整的返回值其实是一个元组，保留第一个元素就行，后面元素是空白。下图即是一个简单的计算图，其中蓝色代表叶子节点，绿色代表根节点，灰色代表要进行的运算，箭头方向为前向传播的方向。
2023-10-20 20:38:07
156
原创 Ubuntu 安装CUDA
可以发现，我这里的NVIDIA显卡驱动版本是 470.141.03，这里可以看到本机最高支持 cuda 11.4 ，也就是说11.4之下的CUDA版本该显卡都可以支持，11.4之上的则不支持。通过上表可以发现，由于自己的NVIDIA显卡驱动版本是 470.141.03，所以自己电脑显卡支持的版本CUDA 11.4及之下。后,终端就会跳出选择界面，回到正常终端模式，并且没有用任何提示。这是正常现象，等待安装完成即可。所以需要根据已有的显卡驱动版本，去。（以CUDA11.4为例）
2023-10-12 16:19:15
8099
1
原创 CenterNet模型复现（超详细）
其他命令:卸载原先的torch、torchaudio和torchvision：代码复现：克隆CenterNet网络：https://github.com/xingyizhou/CenterNet安装COCO数据集：安装所需包：若在其中构建opencv-python错误的话请参考：https://blog.csdn.net/u012331046/article/details/1318029394. 删除CenterNet/src/lib/models/networks下的原始DC
2023-10-12 15:58:26
362
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人