zwv9 commited on
Commit
9095ff2
1 Parent(s): 8c4b798

cudnn no tf32

Browse files
Files changed (1) hide show
  1. wfx.py +1 -1
wfx.py CHANGED
@@ -10,7 +10,7 @@ logger = logging.getLogger('wfx')
10
 
11
  if pv.parse(torch.__version__) >= pv.parse('1.12.0'):
12
  torch.backends.cuda.matmul.allow_tf32 = True
13
- torch.backends.cudnn.allow_tf32 = True # not sure...
14
  logger.info('matching torch version, enabling tf32')
15
 
16
 
 
10
 
11
  if pv.parse(torch.__version__) >= pv.parse('1.12.0'):
12
  torch.backends.cuda.matmul.allow_tf32 = True
13
+ #torch.backends.cudnn.allow_tf32 = True # not sure...
14
  logger.info('matching torch version, enabling tf32')
15
 
16