gpu模式: Caffe::set_mode(Caffe::GPU)
gpu选择:Caffe::SetDevice(int gpuid)
此外,上面两条命令 setmode 以及setdevice 在运行后,会自动占用一些显存,所以setdevice 一定是在setmode 之前,不然会默认占用GPUid0的一部分内存。具体原因不明。
gpu模式: Caffe::set_mode(Caffe::GPU)
gpu选择:Caffe::SetDevice(int gpuid)
此外,上面两条命令 setmode 以及setdevice 在运行后,会自动占用一些显存,所以setdevice 一定是在setmode 之前,不然会默认占用GPUid0的一部分内存。具体原因不明。