命令行常用的方法:
1.跳转到keras.json文件,修改是否采样tensorflow/theano作为后台,
命令: sudo gedit ~/.keras/keras.json
跳转后的页面如下:
(1)theano作为后台:
{
"image_dim_ordering": "th",
"epsilon": 1e-07,
"floatx": "float32",
"image_data_format": "channels_first",
"backend": "theano"
}
(2)tensorflow作为后台:
{
"image_dim_ordering": "tf",
"epsilon": 1e-07,
"floatx": "float32",
"image_data_format": "channels_last",
"backend": "thensorflow"
}
注意:采用tensorflow和theano作为后端时,其通道(RGB)的顺序是不样的。
2.跳转到.theanorc文件中去设置采用的CUDA版本,使用的GPU等
命令:sudo gedit ~/.theanorc
跳转后的界面:
[global]
floatX = float32
device = gpu1 //采用的当前GPU
[cuda]
root = /usr/local/cuda-8.0 //cuda的版本号
3.打开配置CUDA的路径,在该文件内可以手动配置调用那一块GPU(还需研究)
sudo gedit ~/.bashrc
4.查看GPU的配置及使用情况
nvidia-smi
界面如下:
:
5.查看CUDA和CUDNN的版本号:
cuda版本号:cat /usr/local/cuda/version.txt ;
cudnn版本号:cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2