- HiVT源码链接:https://github.com/ZikangZhou/HiVT
1 论文环境配置
论文的环境配置在HiVT源码的README中有写(见下图)
但是有一些步骤需要注意,以下重新叙述一遍安装步骤并标出一些坑:
1.Clone this repository:
git clone https://github.com/ZikangZhou/HiVT.git
cd HiVT
2.Create a conda environment and install the dependencies:
- 先根据显卡型号安装显卡驱动,参照显卡驱动安装。
- 然后根据显卡驱动型号安装对应的 CUDA 版本,参照CUDA安装,这里要注意虽然CUDA版本已经到12.5了,但是pytorch支持的CUDA版本还只到12.1(虽然12.1以上也能支持,但是安装很麻烦),所以建议CUDA版本最高安到12.1。
- 安装与CUDA版本对应的pytorch和cudatoolkit,参照pytorch安装
- 安装python-geometric和python-lightning
pip install torch-geometric==1.7.2
pip install pytorch-lightning==1.5.2
3.安装Argoverse 1 API(详见下面)
2 Argoverse 1 API安装
安装Argoverse 1 API源码https://github.com/argoverse/argoverse-api的步骤来(见下图)
可能会遇到一些问题:
- 报错lapsolver安装错误,这是因为lapsolver只给python3.5-3.6提供了pip安装的途径,我们这里用的是python3.8,需要用源码安装,参照https://github.com/cheind/py-lapsolver。
- 其他问题参照https://blog.csdn.net/m0_48100146/article/details/139543070?ops_request_misc=&request_id=&biz_id=102&utm_term=HiVT&utm_medium=distribute.pc_search_result.none-task-blog-2allsobaiduweb~default-6-139543070.nonecase&spm=1018.2226.3001.4187