AffordaceNet属性学习网络踩坑记录(二)

1、在编译代码网络下的caffe时,要注意是在root权限还是用户权限下,需要在同一权限下执行caffe编译和训练网络。推荐在root目录下执行

2、在运行测试网络时,使用的python路径为代码路径下编译好的caffe。不需要写其他版本caffe的python路径

export PYTHONPATH=/home/qxn/AffordanceNet_ROOTV1/caffe-affordance-net/python:$PYTHONPATH

3、在python环境下确保import caffe不会出错之后再去运行训练命令

4、训练命令示例:

./experiments/scripts/faster_rcnn_end2end.sh 0 VGG16 pascal_voc

5、在系统使用多个python版本时,要确定自己编译caffe的python版本与运行训练命令的版本一致,训练命令可以用以下命令替代,从而指定python版本。

我编译caffe用的是Python3.8,系统自带Python2.7。系统默认python版本也是2.7,但是我编译是在python3下编译的,所以解决方法有两种,第一,使用下面的代码直接指定python版本。

python3 train_net.py --gpu 0 --solver /home/qxn/AffordanceNet_ROOTV1/models/pascal_voc/VGG16/faster_rcnn_end2end/solver.prototxt --weights /home/qxn/AffordanceNet_ROOTV1/data/imagenet_models/VGG16.v2.caffemodel --imdb voc_2012_train --iters 200000 --cfg /home/qxn/AffordanceNet_ROOTV1/experiments/cfgs/faster_rcnn_end2end.yml

第二,修改系统默认python版本,命令如下:

sudo update-alternatives --install /usr/bin/python python /usr/bin/python2 100

 sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150

此命令的作用在于设置Python3的优先级大于Python2.

6、查看调用python的路径

python
import sys
sys.path

注意在sudo下和普通用户下,二者输出一样。

7、注意/etc/profile下关于Python的变量与~/.bashrc下Python的变量一致,最好不要将Python的变量加到profile下。

8、出现cudnn(4 vs 0)的问题,首先确认是否是显存爆了,如果没有,在root下执行训练命令,如果依然出错表示不是权限的问题,可能是cuda和cudnn版本过高,与代码不匹配的问题,可以尝试以下CUDA9

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值