Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Lỗi khi load model seq2seq trên cuda12  #127

@thienbui

Description

@thienbui

Chào Quốc, mình mới nâng cấp cuda12 (sử dụng torch 2.2.2 và torchvision 0.17.2 ) load weight seq2seq bị lỗi như sau:

RuntimeError: cuDNN version incompatibility: PyTorch was compiled against (8, 9, 2) but found runtime version (8, 5, 0). PyTorch already comes bundled with cuDNN. One option to resolving this error is to ensure PyTorch can find the bundled cuDNN. Looks like your LD_LIBRARY_PATH contains incompatible version of cudnn. Please either remove it from the path or install cudnn (8, 9, 2)

trong khi đó load weight của transformer ko lỗi.

cuda và cudnn của mình như này:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Fri_Jan__6_16:45:21_PST_2023
Cuda compilation tools, release 12.0, V12.0.140
Build cuda_12.0.r12.0/compiler.32267302_0

Có thể giúp mình đc ko?
Cảm ơn.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions