2018년 5월 1일 화요일

How to use another gpu when the first gpu is out of memory

tip: https://github.com/BVLC/caffe/issues/440
http://kawahara.ca/caffe-how-to-specify-which-gpu-to-use-in-pycaffe/

before initializing gpu, set the gpu_id first

댓글 없음:

댓글 쓰기