在极客云的GPU 上成功让textcnn跑起来

期间遇到的问题我来整理一下

 

  1. 版本不兼容,注意,我在本机上做实验,用的tf和keras分别是1.40.和2.1.5,在云服务器上一开始爆出了“sofmax的axis的问题,解决方法就是修改版本tf修改为1.4.0keras修改为2.0.8,结果不再报错

 

我以前的tensorflow版本是1.4.1和keras版本2.1.5。

降级到tensorflow版本1.4.0和keras版本2.0.8。

  1. PKL文件打开失败,修改下权限就好了”chmod -rmxrmmx  issureFile"
  2. 把模型的输入数据都复制到/input文件夹,"cp空格/data/xxx/mydata.pkl空格/input
  3. 用GPU训练模型要确保有“tensorflow-gpu"这个模块啊,之前试了多次”CUDA_VISIBLE_DEVICES=0" 都没解决 但学会了查看GPU的指令“nvidia-smi"
  4. 注意文件的路径,要把程序中涉及到文件读入,和存入的路径,都分别设为”/input/inputdata.pkl"和“/data/xxxx/output_data.xx"    

textcnn代码

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值