cuda版本

nvcc --version

cudnn版本

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\include\cudnn_version.h
# C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\【改为你的版本】\include\cudnn_version.h

image.png
这样的话就是8.1.1

补充:如果没找到cudnn_version.h,去cudnn.h里边找

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐