Open hailiang-wang opened 4 years ago
export PYTHONUNBUFFERED=1
export PATH=/opt/miniconda3/envs/venv-py3.5/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
./shell/train_douban.sh
报告错误,
TypeError: softmax() got an unexpected keyword argument 'axis'
解决方案:https://github.com/keras-team/keras/issues/9621#issuecomment-411816766
vi /opt/miniconda3/envs/venv-py3.5/lib/python3.5/site-packages/keras/backend/tensorflow_backend.py
修改,
版本
conda 4.8.2
Python 3.5.5 :: Anaconda, Inc. CUDA 8.0, cudnn 6.0 Ubuntu 16.04
All Python packages
pip freeze