分析-nvidia-smi 中的CUDA 版本与 nvcc不一致
1、CUDA 有两种API，分别是 运行时 API 和 驱动API，即所谓的 Runtime API 与 Driver API。 nvidia-smi 的结果除了有 GPU 驱动版本型号，还有 CU ...
花费 33 ms
pytorch 调用forward 的具体流程
forward方法的具体流程： 以一个Module为例：1. 调用module的call方法2. module的call里面调用module的forward方法3. forward里面如果碰到Mod ...
darknet源码学习
darknet是一个较为轻型的完全基于C与CUDA的开源深度学习框架，其主要特点就是容易安装，没有任何依赖项（OpenCV都可以不用），移植性非常好，支持CPU与GPU两种计算方式。1、test源码( ...
pytorch 加载训练好的模型做inference
前提： 模型参数和结构是分别保存的 1、 构建模型(# load model graph) model = MODEL() 2、加载模型参数(# load model st ...
Ubuntu16.04 安装cuda9.0 cudnn 7.0.5
参考网址：https://blog.csdn.net/zhuangwu116/article/details/81063234 （1）下载安装文件: 下载cuda9.0 runfile 文件 下载 ...
pytorch dataloader 取batch_size时候 出现bug
1、RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 342 and 2 ...
pytorch 常用问题解决
1、RuntimeError: cuda runtime erorr (77): an illegal memory access was encountered at 在使用命令前面加上CUD ...
关于Tensorflow 加载和使用多个模型的方式
在Tensorflow中，所有操作对象都包装到相应的Session中的，所以想要使用不同的模型就需要将这些模型加载到不同的Session中并在使用的时候申明是哪个Session，从而避免由于Ses ...
pytorch模型训练加速tricks
1、学习率设置策略 Pytorch 已经实现了两种方法：「torch.optim.lr_scheduler.CyclicLR」和「torch.optim.lr_scheduler.OneCycleL ...
安装完cuda之后 安装cudnn
nvcc --version（自己安装的cuda版本） 版本需要和nvidia-smi查看到的cuda版本（nvidia驱动对应的cuda版本）一致 1、解压：tar zxvf cudnn-10.1 ...