是的,通常情况下cudnn需要安装在cuda的目录下才能被正确识别和使用。当安装cuda时,会在系统环境变量中添加CUDA_HOME变量,指向cuda的安装目录。而cudnn的安装路径通常是在CUDA_HOME目录下的子目录中。因此,建议将cudnn安装到CUDA_HOME目录下的子目录中,这样就可以在tensorflow中正确引用cudnn了。

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐