配置这个文件的目的是,可以切换cpu和gpu的使用
步骤一:ctrl+alt+t打开终端,打开theanorc文件
sudo gedit ~/.theanorc
空白文件中写入:
[global]
floatX=float32
device=cpu
步骤二:修改内容在上边基础上修改
1、使用gpu加速时,把上边device修改,同时添加新的内容
device=gpu
[cuda]
root=/usr/local/cuda-8.0
2、使用cudnn加速时,继续添加
[nvcc]
fastmath =True
optimizer_including=cudnn
3、想要保证使用cuda运行过程中参数不随机时候添加设置
[dnn.conv]
algo_bwd_filter =deterministic
algo_bwd_data =deterministic
4、并行操作时配置添加
默认情况下,显卡显存占满,只允许一个程序运行如下所示:初始化时调用30%的显存,可最大允许三个程序同时运行。
[lib]
cnmem=0.3
网友评论