关于为yolo配置环境,安装显卡驱动,CUDA,cudnn,torch,torchaudio,torchvision,还有python版本的对应关系。

出现该问题的原因:最近换电脑,手头三个电脑显卡型号英伟达(GT730 1060 4070),驱动显卡配置python环境,这些版本真的搞得我头都大了。

目的:为了配置好环境运行yolo

具体的操作步骤

一、安装显卡驱动(网址:官方驱动 | NVIDIA

 在此处按自己的具体配置情况填写,该网站会自动识别,但是未必会与你的显卡相匹配,所以还是自己检查填写一下。然后点击搜索下载,下载好文件就能直接安装了。

 具体的安装方法可参考这篇文章【Windows】安装NVIDIA驱动 / 更新驱动_win10 安装nvidia_摇曳的树的博客-CSDN博客,这里不赘述。

然后我们cmd打开win10 的命令窗口,输入nvidia-smi(注意中间没有空格)。

右上角写的CUDA Version:12.2就是我们能够安装的CUDA的最高版本。

总结一下:第一步我们安装了显卡的驱动,并且我们获知了我们能够安装CUDA的最高版本(实际可能不会安装最新的版本,原因是pytorch版本开发是稍迟于CUDA版本的,后续有说到)。

二 、选择CUDA与(torch,torchaudio,torchvision)的版本对应关系。

pytorch下载网站:官网下载:PyTorch

pytorch国内下载镜像:Index of /anaconda/cloud/pytorch/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

cuda下载网站:CUDA Toolkit Archive | NVIDIA Developer

我的思路是,由于pytorch版本是落后于CUDA的,并且第一步我们也获得的我们的显卡支持的最高级的CUDA版本,这时候我们就找到在我们电脑能安装的(torch,torchaudio,torchvision)版本就好,这里官网能够给我们完美推荐。并且我们能看到(torch,torchaudio,torchvision)匹配的CUDA版本,找到对应版本安装就好。

 选择好自己的配置后,复制下边代码(anaconda或者命令提示符运行),在运行前先安装自己在这里选择匹配pytorch的cuda版本。接下来第三步讲解CUDA安装方式,第四步讲(torch,torchaudio,torchvision)的安装

三 、安装CUDA并验证安装成功

留坑更新

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值