环境配置
在环境配置之前,首先要进入主板BIOS关闭安全启动,不然后面装驱动会报错
CUDA安装
NVIDIA驱动安装
ubuntu20.04 安装NVIDIA驱动很容易,只只需要打开系统设置->软件和更新->附加驱动->选择NVIDIA驱动->应用更改。该界面会自动根据电脑上的GPU显示推荐的NVIDIA显卡驱动。
NVIDIA(英伟达)显卡驱动安装完成后,在终端输入nvidia-smi
输出如下图所示的结果就表示安装成功了。下图中绿色框内表明,安装的驱动支持的CUDA最高版本为12.0(注意这里需要重新启动电脑)。
安装CUDA
先进一下pytorch官网看一下cuda和pytorch的搭配
Previous PyTorch Versions | PyTorch
所以我选择了CUDA 11.8版本
CUDA官网CUDA Toolkit Archive | NVIDIA Developer
参考文章CSDN中的大佬文章:2023年最新Ubuntu安装pytorch教程
2023年最新Ubuntu安装pytorch教程-CSDN博客
这个地方用到了vim编辑器,安装一下才能用
这里vim编辑器,键盘输入i插入,esc退出编辑,输入:wq保存编辑
跟着教程来安装就好。
安装Anaconda
Unleash AI Innovation and Value | Anaconda
这里不要这样安装
我的做法是将安装包放在/home/ming(主目录)下,右键在终端中打开
bash anaconda3.xxxxx.sh(用Tab键补全)
这里按回车慢一点,不然会漏掉确认
这里看一下安装路径,不要像图片一样安装在root下,如果显示安装在/home/name/anaconda3就是对的。
创建anaconda环境
按照教程来就行
安装pytorch
这边一定要在激活dl环境的条件下安装
就是终端要名称前会带上dl
跟着教程到这就结束了
pycharm安装
跟着大佬的文章安装
在Ubuntu中安装并配置Pycharm教程_ubuntu安装pycharm-CSDN博客
主要是要在配置中选择我们安装了torch的dl环境
至此DANNet所需环境就配置好了
代码复现
下载好解压,用pycharm打开
点进去
测试的话先下载这个包
下载好了解压,最好跟Dannet的代码放在一个文件夹中
根据他的Testing还需要进入这个网站下载一个包
https://www.dropbox.com/s/fmlq806p2wqf311/trained_models.zip?dl=0
Step1: download the [trained models](https://www.dropbox.com/s/fmlq806p2wqf311/trained_models.zip?dl=0) and put it in the root.
Step2: change the data and model paths in configs/test_config.py
Step3: run "python evaluation.py"
Step4: run "python compute_iou.py"
Step1
下载好这两个包,解压
Step2
pycharm打开项目
这四个目录要改
第一个是你下载的Dark_Zurich_Val_annon放的位置
可以在文件夹中找到.pth文件使用ctrl+L可以复制文件路径
第二个是DAnnrt-main文件夹dataset目录,也是找找就能找到
第三个是下载的trained_models解压后的_psp.pth文件的路径
第四个也是下载的trained_models解压后_light.pth文件的路径
Step3
运行 "python evaluation.py"文件
Step4
运行 "python compute_iou.py"文件
这里我报过错,一次是说np.int不太行,查资料说是版本更新np.xxx的变量类型被抛弃了。
我的解决方式是卸载原有的numpy,安装其他版本的,可以看大佬文章
还有一个坑就是,这里要卸载与重装的时候,要记得先激活dl环境
conda activate dl
处理完这个我还报过错
意识是一个文件notfound
解决办法是依次点击错误,会有一个跳转到另一个.py文件中
notfound的是result文件中的某个文件
需要手动将一个/path/to/goxxx的路径改成result文件夹的目录
至此,代码的Testing部分就算是成功复现了。
一些感悟
从零接触Ubuntu系统,第一次复现大牛们的代码。复现工作是我在2024年春节期间进行的,在全家欢度新春的时候,只有我因此心事重重,乌云笼罩,有一种叫天天不应,叫地地不灵的无力感。但好在,精诚所至,金石为开,我真的做到了。虽然前路还会遇到无数挑战,但我终会想起成功跑起来的那个下午明媚的阳光。
继续加油吧!