3D-R2N2 环境配置及训练

xiaoxiao2021-02-28  75

1、 old gpu-backend

运行测试用例:vim ~/.theanorc 将device=cuda×

运行训练脚本:更改脚本res_gru_net.sh 第28行 device=gpu

2、compiling train function error

确认cudnn版本为5.1 cuda8.0

Theano 卸载:进入虚拟环境,输入: pip uninstall theano

转载请注明原文地址: https://www.6miu.com/read-1450342.html

最新回复(0)