cudnn no tf32
Browse files
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 |
|