Ultra-Fast-Lane-Detection-v2代码实现
文章目录
前言
第一次写,别骂我
一、背景介绍
做毕设中,需要先跑几个模型起来
paper:https://arxiv.org/pdf/2206.07389.pdf
代码链接:https://github.com/cfzd/Ultra-Fast-Lane-Detection-v2
论文解读(我的解读,正常人别看):暂时还没有
二、环境搭建
1.克隆项目
下载代码(要是这个代码不太行就直接下载哈):
git clone https://github.com/cfzd/Ultra-Fast-Lane-Detection-V2
cd Ultra-Fast-Lane-Detection-V2
2.创建 conda 虚拟环境并将其激活
conda create -n lane-det python=3.7 -y
conda activate lane-det
3.安装依赖项
# If you dont have pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
注意:这里如果这条代码没法直接下载就是cuda和pytorch版本不对应
处理方法:
1. 查看版本:方式一:
conda list
可以看到pytorch版本和cuda的版本
方式二:在python代码中验证查看print(torch.cuda.is_available()) # 查看CUDA是否可用 print(torch.cuda.device_count()) # 查看可用的CUDA数量 print(torch