在C++中使用OpenMP庫來設(shè)置線程數(shù),可以通過設(shè)置環(huán)境變量OMP_NUM_THREADS
來控制線程數(shù)。
#include <omp.h>
#include <iostream>
int main() {
// 設(shè)置線程數(shù)為4
omp_set_num_threads(4);
#pragma omp parallel
{
int thread_id = omp_get_thread_num();
std::cout << "Hello from thread " << thread_id << std::endl;
}
return 0;
}
另外,可以通過在運行程序時設(shè)置OMP_NUM_THREADS
環(huán)境變量來控制線程數(shù),例如在Linux系統(tǒng)下可以使用以下命令:
export OMP_NUM_THREADS=4
./your_program