> 文章列表 > 在使用ray tune时出现cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

在使用ray tune时出现cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

在使用ray tune时出现cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

  • 使用ray tune进行调参时,出现这个状况很显然是需要调整一下这个参数resources_per_trial
    result = tune.run(train_cifar,resources_per_trial={"cpu": 12, "gpu": 1},	# 这个config=config,num_samples=num_samples,scheduler=scheduler,local_dir=absolute_path+"/MLP/ray_results",progress_reporter=reporter)
  • 这个参数负责分配多少cpu和gpu资源,默认是1 CPU and 0 GPUs,但是大多数电脑都可以承受更大的cpu资源以及GPU运算。因此如果model需要gpu要设置上这个参数,
  • 一些人会把GPU设置成0.5也是可以的