NeuralTalk2模型笔记

本文记录了在安装和训练NeuralTalk2模型过程中遇到的问题及解决方案,包括torch安装、cudnn版本匹配、hdf5降级、模型运行、数据准备和训练过程。最终在有限的迭代次数下,测试准确率为70%。
摘要由CSDN通过智能技术生成

模型地址:

https://github.com/karpathy/neuraltalk2

遇到的坑:

1.torch安装过程中,文件尺寸不符

解决办法:分析缺失哪些包,手动下载

2.cudnn版本不对

本机环境 ubuntu17.10+cuda8

cudnn 一开始下的是6.0版本,后来降级为5.1版本

5.1下载地址

链接:http://pan.baidu.com/s/1c984eq 密码:wai9

安装

https://www.linuxidc.com/Linux/2016-12/138906.htm

环境配置

export CUDNN_PATH="/usr/local/cuda/lib64/libcudnn.so.5"

3.hdf5版本不对

默认下载hdf5-1.10 应降级为

hdf5-1.8.20

 解压。进入目录,
先 ./configure --prefix=/usr/local/hdf5-1.8.20
而后执行 make , make check , sudo make install , sudo make install check
    安装完毕后,修改
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值