深度学习
文章平均质量分 83
文山湖的猫
Long may the Sun shine!
展开
-
NVIDIA Jetson Xavier NX入门(3)——安装miniforge和Pytorch
1 安装miniforge1.1 miniforge简介conda是一个开源的包、环境管理器,可以用于在同一个机器上安装不同版本的软件包及其依赖,并能够在不同的环境之间切换。搞深度学习的应该都十分熟悉anaconda,但是NVIDIA Jetson Xavier NX是arm架构的,anaconda及其精简版miniconda并不支持arm64架构。现在主流的CPU架构分为Intel的x86/x64架构和ARM的ARM/ARM64两种,平常用的电脑大部分都是x86/x64的(苹果除外),Xavier使用原创 2021-08-16 20:10:45 · 7003 阅读 · 5 评论 -
NVIDIA Jetson Xavier NX入门(2)——开机设置和远程登录
1 开机设置如果系统镜像烧录成功的话,插入SD卡后,NVIDIA Jetson Xavier NX就可以直接运行了。系统是定制的Ubuntu18.04,开机后同意协议,一直点击continue就可以了。语言建议选择English,避免后期因为语言出现什么问题,有时候就是这么不讲理的。 Xavier官方系统里自带cuda10.2,但是需要手动添加进环境路径,否则在之后的使用中会出现找不到cuda的错误。右键桌面空白处,选择“Open Terminal”,呼出终端,输入以下代码:sudo vim原创 2021-08-16 20:09:48 · 6522 阅读 · 1 评论 -
pytoch使用nn.DataParallel导致Missing key(s) in state_dict错误
1.nn.DataParallelnn.DataParallel是pytorch使用多gpu训练时所使用的方法,但是使用nn.DataParallel之后,模型的读取就会有所不同。最常见的情况就是使用的预训练模型并不是在多张gpu上训练得来的,没有使用nn.DataParallel包装,但现在想要使用这个预训练模型就不能直接用下面的代码model.load_state_dict(torch.load(save_path))否则就会报以下错误Missing key(s) in state_dict:原创 2021-03-05 21:09:26 · 1582 阅读 · 1 评论