cvpr_2018文章的实验
配置
python 3.6
keras=1.1.0
Theano = 1.0.2(pip install)
Pygpu = 0.7.5(conda install)
Tensorflow,tensorflow-gpu=2.2.0
步骤
Conda create -n my_new_env python=3.6
Conda activate my_new_env
pip install Theano==1.0.2
conda install pygpu=0.7.5
python check_gpu.py
修改keras后端
进入个人目录的主目录下:
cd ~
cd .keras
vim keras.json
点击键盘上的i,移动光标,然后将backend修改成theano。按一下esc,再按:wq,即保存并退出。