CUDA随机数库的使用
深度学习过程中经常需要使用随机数, CUDA提供了一个高性能的随机函数库,如果要使用的话,可能有些坑你会遇到。具体步骤如下:
#include <curand.h>
- 编写代码
do {
int b = 0;
curandCreateGenerator(&gen, CURAND_RNG_PSEUDO_DEFAULT);
int a = b != CURAND_STATUS_SUCCESS;
if (a) {
/*
printf("Error at %s:%d\n", __FILE__, __LINE__);
return EXIT_FAILURE;
*/
}当你按contro+f5,你可能遇到无法找到符号的错误,那是因为没有在链接器上引入curand模块,具体操作如下: