RT-DETR源码学习bug记录
事情是这样的,我最近想学习RT-DETR的源码,那就开始吧!
1. 找到官网,找到pytorch版本。
https://github.com/lyuwenyu/RT-DETR/tree/main
2.只想下载一个子目录,方法:
https://download-directory.github.io/,然后在搜索框填入https://github.com/lyuwenyu/RT-DETR/tree/main/rtdetr_pytorch,回车即可。
3. 下载好之后,拖入服务器。
conda create -n rtdetr python=3.8
conda activate rtdetr
cd /home/cj/workspace/python/studyModel
git clone https://github.com/lyuwenyu/RT-DETR.git
cd /home/cj/workspace/python/studyModel/RT-DETR/rtdetr_pytorch
pip install -r requirements.txt
# install pytorch of the required version using conda
# makes sure correct version for your cuda version is installed
conda install pytorch==2.0.1 pytorch-cuda=12.1 -c pytorch -c nvidia
pip install -r requirements.txt
pip install transformers
4.这里报错卡了2天,扫不到包,原来要设为根目录