mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Checks Compute Capability and forcibly disables Tensor Cores for CC < 7.0

Browse Source
This commit is contained in:
AlexeyAB
2019-02-04 23:28:40 +03:00
parent f7cb538b32
commit 5446d19576
3 changed files with 29 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/parser.c
View File

@ -671,8 +671,8 @@ void parse_net_options(list *options, network *net)
net->policy = get_policy(policy_s);
net->burn_in = option_find_int_quiet(options, "burn_in", 0);
#ifdef CUDNN_HALF
//net->burn_in = 0;
net->cudnn_half = 1;
if(get_gpu_compute_capability(net->gpu_index) >= 7000) net->cudnn_half = 1;
else net->cudnn_half = 0;
#endif
if(net->policy == STEP){
net->step = option_find_int(options, "step", 1);