Dsp Tian

摘要： 如果一个机器上有多个gpu，可以使用多gpu训练。 一般数据量和模型比较大的时候训练速度会有明显的提升，模型和数据比较小的时候反而可能因为数据通信原因导致性能下降。 下面是一个简单的例子： import time import torch import torchvision.models from 阅读全文

摘要： 有时候pip下载安装特别慢，可以考虑更换下载源： 清华源： pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple 阿里源： pip config set global.index-url https://mi 阅读全文

摘要： 主要用了两个库：pipdeptree和pyecharts。 pipdeptree用来解析所有库的依赖关系。 pyecharts用来显示依赖的网络关系。 通常的方法应该是通过输入下面命令： pipdeptree --json-tree > pkg.json 得到关系树文件，然后解析该文件并在pyech 阅读全文

摘要： 混合精度训练通常会结合使用单精度浮点数（float32）和半精度浮点数（float16），以提高训练效率和减少内存占用。 代码中关键在于两个地方： 1. 在with autocast():下，模型的前向传播和损失计算放在自动混合精度加速环境中进行。 2. 使用scaler对象进行混合精度训练的管理。 阅读全文

摘要： 昨天一切正常，今天ubuntu启动突然卡住了，一直停在黑屏光标界面。 在网上看到原因可能和nvidia驱动有关。 输入nvidia-smi报出题目中的错误。 一般这种错误都是在刚安装nvidia驱动后第一次启动可能出现，见我这篇文章。 不过这次是突然出现的，不是安装驱动后出现的，所以不是安装nvid 阅读全文

摘要： Schur 补是一种矩阵分解方法，通过将一个大的矩阵分解为几个较小的矩阵来简化计算，通常能够提高矩阵求逆的速度。 对于形如下面的矩阵： 可以把矩阵划分为左上、右上、左下、右下四个分块矩阵。 得到矩阵： 根据A和D的奇异性，可以分两种情况。 如果A可逆，则有： 如果D可逆，则有： 公式两边同时求逆，得 阅读全文

摘要： Q-Learning是一种基于值函数的强化学习算法，这里用该算法解决走迷宫问题。 算法步骤如下： 1. 初始化 Q 表：每个表格对应状态动作的 Q 值。这里就是一个H*W*4的表，4代表上下左右四个动作。 2. 选择动作： 根据 Q 表格选择最优动作或者以一定概率随机选择动作。 3. 执行动作，得到 阅读全文

摘要： 编译有些python库时可能会报这个错。 解决方法是： 将setup.py中的“cmdclass={'build_ext': BuildExtension}”改为“cmdclass={'build_ext': BuildExtension.with_options(use_ninja=False)} 阅读全文

摘要： 一般出现这种问题都是cuda、pytorch或vs版本没配对上。 这里出现问题的文件为torch中的cast.h。 我看网上很多给的解决方法是修改下面代码： template <template <typename...> class Tuple, typename... Ts> class tup 阅读全文

摘要： 下面这个命令： ps -e | grep gsd | awk '{print $1}' | xargs -I{} sh -c 'cat /proc/{}/status && echo " "' > status.txt 用于获取系统中所有进程名称中包含"gsd"的进程的状态，并将结果保存到名为"st 阅读全文