Skip to content
This repository was archived by the owner on Dec 9, 2024. It is now read-only.
This repository was archived by the owner on Dec 9, 2024. It is now read-only.

resnet50 --use_fp16 error: cuDNN launch failure : input shape ([128,112,112,64]) #518

@anefischer

Description

@anefischer

3090 and following versions:
Windows 10
python 3.9.5
tensorflow 2.5
CUDA 11.2.2 (path set)
CuDNN 8.1

fp32 works:
python tf_cnn_benchmarks.py --num_gpus=1 --batch_size=128 --model=resnet50 --variable_update=parameter_server

fp16 not:
python tf_cnn_benchmarks.py --num_gpus=1 --batch_size=128 --model=resnet50 --variable_update=parameter_server --use_fp16
error:
Internal: cuDNN launch failure : input shape ([128,112,112,64])

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions