安装torch成功但是编译找不到

跟着学yolov7,但是我的torch是安装好了，但是编的时候不行，跟着python的提示做更新了也不行，跟着他下载一直卡住，但是按教程我的torch是跟你一样的，可能是我的CUDA是11.4版本的，但是我没有找到对应的版本，所以我torch下了11.3版本的

  • 这篇博客: 一文搞懂CUDA问题，警告：买不起显卡的勿进中的 第三条线 Torch与CUDA版本对应 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:

  • 图片

    这个图网上摘抄的，顺便说明了与python和torchvision的版本关系，大家只要确定好torch版本，其他的都可以确定了。

    第四条线 CUDNN与CUDA

    之所以把这条线放到最后，是因为即使不使用cudnn也可以进行正常的跑代码了。

    那如何匹配CUDA的版本呢？

    参考地址：cuDNN Archive | NVIDIA Developer

    图片

    这里的cudann最后都有一个for CUDA的版本号，大家根据版本号找对应的下载就行，下载好之后解压复制到cuda的文件下，就可以完成安装了，是不是很方便

    最后给大家附上几个链接：

    cudnn下载：cuDNN Archive | NVIDIA Developer

    pytorch官网：Previous PyTorch Versions | PyTorch

    显卡驱动下载：官方 GeForce 驱动程序 | NVIDIA

    cudnn的安装与卸载：CUDA与cuDNN - 简书 (jianshu.com)

    最后欢迎大家关注，点赞~

    下期分享一下怎么用显卡跑gazebo仿真，体验丝滑的赶脚~