在Keras中将cpu切换为gpu

xiaoxiao2021-02-28  129

在终端中输入gedit ~/.theanorc 这时文档应是空白的,在文档中写入:

[global] device=gpu floatX=float32 [dnn.conv] algo_bwd_filter = deterministic algo_bwd_data = deterministic [cuda] root=/usr/local/cuda-8.0 [lib] cnmem=0.3 [nvcc] fastmath = True optimizer_including=cudnn

参考:http://blog.csdn.net/sinat_31824577/article/details/53140406

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

最新回复(0)