yoloV5环境配置及pytorch配置

yoloV5中遇到的坑

使用pip install -r requirement.txt下载依赖,所下载的是torch而非pytorch,且依然能用cpu训练

解决方法

自行下载pytorch和其他依赖


pytorch安装步骤:
  1. 安装cudn显卡驱动(nivida官网下载)
  2. anaconda下载cudatoolkit,cudnn(依赖cudatoolkit),pytorch(依赖cudatoolkit、cuda和python),torchvision和torchaudio(依赖)

例:cudatoolkit _11.3.1 cudnn-8.2.1-cuda11.3_0 pytorch-1.12.0-py3.9_cuda11.3_cudnn8.0 torchvision-0.13.0-py3.9_cu113

conda install cudatoolkit =11.3.1 cudnn=8.2.1 pytorch=1.12.0 torchvision=0.13.0


** 有时候版本对应但是仍然无法下载,比如上述例子中pytorch-1.12.1-py3.9_cuda11.3_cudnn8.0无法下载,但1.12.0可以下载 **

  • 11
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值