- 博客(8)
- 收藏
- 关注
RSS订阅原创 训练时是否使用显卡和怎么使用多张显卡的问题
前言：修改代码时心生疑问，故写此文章，以便参考。model.to(device)和model.cuda()的区别：.to(device) 可以指定CPU 或者GPUdevice = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") # 单GPU或者CPUmodel.to(device)#如果是多GPUif torch.cuda.device_count() > 1: model = nn.DataP..
2022-05-08 18:20:08
3215
3
原创 使用labelImg对自己的数据集进行标注
步骤1.打开命令行输入pip install labelImg若需要更新按照步骤操作即可2.创建基本目录3.在当前窗口右击鼠标+Shift，点击在此处打开powershell窗口之后输入 labelImg ./image ./classes.txt./image表示存放所需标注图片的目录，./classes.txt存放表示类别信息的目录4.设置存放标注信息的目录，可以在标注后设置定位识别难度xm文件信息...
2022-04-15 10:55:05
1098
转载 显卡驱动报错怎么拯救
显卡驱动报错：NVIDIA-SMI has failed because it couldn’t communicate with the NVIDIA driver.感谢大哥
2021-12-29 11:13:14
472
原创 Pytorch开发环境安装
PyTorch简介：PyTorch的前身是Torch，其底层和Torch框架一样，但是使用Python重新写了很多内容，不仅更加灵活，支持动态图，而且提供了Python接口。Pytorch作为当下最火热，发展势头最猛的深度学习框架。PyTorch相对于Tensorflow更易于上手，使得初学者可以更方便的了解深度学习。下面我们来学习PyTorch以及相关软件的开发环境安装。虽然安装顺序并没有要求，但我在如下安装顺序下并没有报错或产生其他异常，所以建议按照以下顺序安装软件。安装Python3.7推荐安
2021-10-27 22:24:47
415
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人