- 博客(6)
- 收藏
- 关注
RSS订阅原创 深度学习，在含有分布式训练的代码中如何指定显卡进行训练
1.修改gpu_number，我的代码是直接指定args.gpu=‘指定的显卡序号’，例如：args.gpu=3。2.需要设置device 参数。根据第一点的例子来看，就需要设置device=cuda:3。这两个参数必须一样，否则就会出现分布式数据并行（DDP）训练的推理报错。
2024-10-04 16:45:56
123
原创 linux第一次安装显卡驱动
重启电脑，再次检查nouveau是否禁用。4.deb文件安装：sudo dpkg -i NVIDIA...deb --no-x-check。lsmod | grep nouveau 来查看nouveau是否禁用，没禁用必须禁用。方法：sudo gedit /etc/modprobe.d/blacklist.conf。1.通常认为装系统自带的驱动不好用，建议去官网安装适合自己显卡的最新的驱动。6.重启终端，输入nvidia-smi，此时可以看到安装成功！sh文件安装：sudo sh ./xxxxx。
2024-09-16 10:35:21
356
1
原创 git clone --branch master --single-branch
这条命令用于下载github中项目的特定分支。修改master为main，就是下载main分支。
2024-09-16 10:23:12
159
原创 MNIST实现的多分类的简单神经网络
本内容主要来自于B站上刘老师的课程：《PyTorch深度学习实践》完结合集_哔哩哔哩_bilibili在观看了视频第九讲之后，将老师视频中的代码照着写了下来。过程中遇到了一些问题，总结如下：1.神经网络整体流程代码不能随便改变各个板块的顺序，不然就会报错。如果下面的程序会用到一些载入的数据或则设定的参数，那么这些数据和参数一定要在最开始就写出来。识别程序的逻辑是由上到下的。2.如果代码只跑了一个progress就停止了，检查代码也没什么问题。建议重新启动电脑之后再试。我之前因为跑了其他程序，..
2022-03-14 15:33:21
781
原创 opencv，python，pil计算图片内存占有量（Kb）简单大概的计算方法
本文主要是转载一个博主的内容。今天在完成老师布置的作业时，有一个内容是展示压缩之后的图片的大小。我首先就使用了img.shape发现压缩前后的shape是一样的，并不是我想要的值。于是又使用了img.size发现压缩前后的size值是一样的，毕竟size = shape的长宽相乘再乘以通道数。于是去搜索展示内存大小的代码，换了好几次关键词。也找到了好几个版本。这里展示我找到的我刚好能用到的版本，供大家参考。用我这个小白的口气来说就是，原理就是把图片转换成了一种可以计算长度的格式，然后
2021-10-04 22:01:58
2008
原创 opencv中sort函数的用法（个人总结）
相信通过这张图，就能大概理解sort函数的一般基础用法。后面输入数字为正负或者0，都会输出不一样的序列。设置为0的情况居多，他会把矩阵中单个元素里面的数值从小到大排列，元素的顺序不会改变。而正负数，则会让元素里面的数值互相移动。换而言之，为正负数时，进行列变化。...
2021-09-26 16:07:09
1264
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人