2016-05-31 4 views
2

우리가 알다시피 mxnet에서 multiplegpu에 대한 개념이 있는데, command에 -gpu를 지정할 때입니다. 그리고 gpu를 지정하지 않으면 cpu에서 실행합니다. 얼마나 많은 cpus가합니까? 여러 CPU를 지정할 수 있습니까? (파이썬은 매우 유사하다에서, R 버전)은 mxnet의 multiplecpu에 대한 개념입니다.

답변

1

다음과 같은 코드를 사용하여 여러 개의 CPU를 사용할 수 있습니다

devices = lapply(1:2, function(i) { 
    mx.cpu(i) 
}) 

그리고 기차 평소와 같이 네트워크. 또한 MKL 라이브러리가있는 경우 시스템은 자동으로 모든 코어를 계산합니다.

0

몇 가지 방법이 있습니다.

좋은 BLAS 라이브러리로 MXNet을 컴파일하면,이 연산은 사용 가능한 모든 CPU 코어를 사용하게됩니다.

또한 환경 변수 MXNET_CPU_WORKER_NTHREADS을 통해 CPU 작업자 스레드 수를 지정할 수 있습니다. http://mxnet.io/how_to/env_var.html

+0

R에서 MXNET_CPU_WORKER_NTHREADS를 설정하는 방법 – Shiv

관련 문제