- 博客(5)
- 收藏
- 关注
原创 Ubuntu18.04 Nvidia driver更新
2. 确认GPU类型,已安装的nvidia driver以及CUDA。所以修复系统中的软件包依赖问题,或者也可以按这篇文章中的方式处理。输入如下命令后,会打开gedit文档编辑器,弹出GUI界面。使用如下命令,选择其中推荐driver的进行安装。在文档中写入如下两行代码,并保存ctrl+s。这时如果直接安装driver,会报错依赖问题。3. 删除已安装的driver和CUDA。4. 确认可安装的driver版本。1. 进行nouveau的无效化。5. 重启,确认安装成功。
2023-10-16 18:58:45 229 1
原创 COCO数据集用法介绍
所有的annotation数据都保存在instances_val2017.json这一个文件中(这里举的是val集的例子)COCO是Microsoft公司发布的多任务数据集,可以用于目标检测,语义分割,关键点检测等多种任务。目标检测主要用到的是拥有80类的object标注数据。
2023-05-06 16:58:11 123
原创 Win10+940MX显卡配置PyTorch1.0.0的痛苦经历
CUDA8.0支持的最高版本是PyTorch1.0.0,而该版本PyTorch需要在Python3.5~3.7之间的版本下运行。最近重新配置了一台2017年的笔记本电脑,花了好几天才配置好,分享一下惨痛经历。电脑搭载的是GeForce 940MX显卡,对应的最高版本是CUDA8.0。因此需要先在NVIDIA官网下载CUDA8.0+对应的cuDNN并进行安装。由于CUDA版本过于老旧,因此不支持最新版的PyTorch2.0.0。因此选择安装2018年版的Anaconda3-4.3.1。这样终于可以成功运行。
2023-05-05 19:14:04 833 4
原创 CNN手写数字识别——使用MSE误差函数
使用MSE误差函数进行手写字体识别时,需要保证label为one-hot形式可以通过target-transform参数调整label数据为one-hot,也可以在学习过程中调整本文采用后者,代码如下import torchfrom torch import nn, optimfrom torch.autograd import Variablefrom torch.utils.data import DataLoaderfrom torchvision import transfor
2022-04-18 19:40:18 852
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人