qq_33949900的博客

测试用例一、需求二、测试用例三、总结 一、需求 最近想弄清楚神经网络的一些细节，苦于网络代码很长，于是想自己写一段测试用例。 二、测试用例 直接上代码，该段代码能测试tensor经过平均池化后的维度变化，这里可以看到平均池化给的参数是（6,10），输出tensor的维度也是（6,10）。 import torch import torch.nn as nn import torch.nn.functional as F import math class FPN(nn.Module): def

切换cuda版本的命令一、前言二、步骤2.1、删除原来的cuda连接2.2、建立新的cuda连接2.3、查看当前cuda版本三、总结 一、前言 由于不同的代码需要不同cuda依赖，所以常常需要切换，特此记录一下。 二、步骤 2.1、删除原来的cuda连接 sudo rm -rf /usr/local/cuda 2.2、建立新的cuda连接 sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda 2.3、查看当前cuda版本 cd /usr/local stat c

创建软连接1、背景介绍2、方案2.1、创建2.2、删除3、总结 1、背景介绍 我们知道在linux中经常需要复制文件，但是由于内存限制，我们可以选择创建软连接，那么如何创建呢？可以使用下面的命令。 2、方案 2.1、创建 ln -s /usr/cuda /media/EA6XXX/Zhou 其中左边是源文件夹，右边是目标文件夹。 2.2、删除 rm -f xxx 和删除文件夹一样，直接删除即可。 3、总结 仅仅为了记录，下次不用翻别人的博客。 ...

常见技巧总结1、GPU篇1.1、选择用哪块GPU1.2、查看GPU使用情况2、代码细节 1、GPU篇 1.1、选择用哪块GPU import os device = os.environ["CUDA_VISIBLE_DEVICES"] = '0' print(device) 一般笔记本电脑都有两个GPU，一个核显，一个独显，所以还是要设置一下ID号的。 1.2、查看GPU使用情况 watch -n 5 nvidia-smi 该命令是每隔5s查看一次GPU显存占用率，当然一般用在ubuntu上，一般主机

记一次成功的安装1、开始安装1.1、报错集锦1.2、重新安装 1、开始安装 总体上而言参考这篇博客，点击进入博客链接。然而遇到了几个错误。 1.1、报错集锦 1、解决OK无法点击问题的链接，Package Configuration无法点击Ok的问题 2、解决ubuntu卡在clean, / files, / blocks不能进入系统的问题， 点击进入链接。 1.2、重新安装 暂时失败，用参考博客安装容易出错，放弃！ ...