全网第一篇:DRRG代码的具体运行方法(亲测)

前言

经过三个星期的刻苦奋斗,终于成功运行出了DRRG,期间经历了无数的坑,我不敢说我是全网第一个跑出来的(肯定不是),但我绝对是第一个写出来教程的,本人也是一个小白,过程不易如果有帮到您请给一个赞,谢谢大家。前几天写的今天回来补充一点,因为很多东西涉及到系统权限,所以请大家用root权限做,不是用root用户哦。
命令:

然后输入你的密码就行了·,要不可能会报错。

效果图

先上训练图,证明我跑了:

全网第一篇:DRRG代码的具体运行方法(亲测)点击代码网址这部分。有条件的小伙伴可以用谷歌云下载,实在不行我告诉你一种办法去淘宝找代下载。
第三步把数据集放到对应文件夹: 全网第一篇:DRRG代码的具体运行方法(亲测)
这就是代码中的README文件。大家可以自己去看,这里我要说一点,他有错误:GCC必须是8,9是不好使的。
下面我们说软件:首先python版本必须是3.7,这点很重要,因为pytorch只能用1.2.0,其他都不好使所以,python必须是3.7,然后centos自带的python是2,所以即使你安了3.7也是按在主环境里面但是后续安装pytorch的时候你可能不知道他具体安在哪个环境里面。所以我推荐是做一个虚拟环境,虚拟环境管理不易,推荐下载anaconda。下面我们进入下载环节。

安装anaconda

为了照顾初学者我们全程用命令安装:
首先右键打开终端,就是这个:

全网第一篇:DRRG代码的具体运行方法(亲测)
注意:为了安装3.7,我们得先点击一下升级数据:Update index,否则你是找不到3.7的。然后点击Create 制造虚拟环境,点完以后是这样 全网第一篇:DRRG代码的具体运行方法(亲测)
然后, 全网第一篇:DRRG代码的具体运行方法(亲测)
你有了这个就是导入成功了。然后我们运行程序。

安装cuda

在运行程序之前我们要先安装cuda10.1,其他版本不行,切记,因为要和pytorch版本对应,具体方法推荐大神的博客:网上有很多教程,大家自己搜一下,如果你成功禁用了原有驱动,你可以运行以下命令:

这些命令会帮你安装cuda10.1,但是他同样会产生网速问题,所以你要做和anaconda一样的处理。切记完成以后要配置环境变量,然后重启。才能生效,然后做的是升级GCC版本这里有个问题,要求文档说GCC是9.0但是我测试后发现,只有8才好使大家别按错了,这里推荐大神的教程:CentOS完美升级gcc版本方法 – 吴昊博客 https://blog.whsir.com/post-4975.html
注意:同样要配置环境变量重启或者source。版本很重要这里再说一遍!!!

跑代码

接下来我们要去跑代码,首先我们要改tool下的train_CTW1500.sh文件,这里主要改两点,batch_size 和你的GPU有关他越大对你GPU要求越高,我设置为2,还有一个参数num_workers他和你的CPU有关,cpu核数越多打就可以设置越大,我设置为4,不要太大否则会卡死,初始文件是没有的,你要自己加上,在末尾后空格加,还有一点不用改但是很重要,gpu的0不是数量是名字,是说第一块gpu的名字叫0,以此类推:

否则会默认为8,可能和卡死,然后保存。
接着我们开始跑:
第一步打开terminal在pycharm的左下角点开它,应该是这样的:

全网第一篇:DRRG代码的具体运行方法(亲测)

首先我们要进入代码文件夹,在$后面输入:

然后我们安装库,如果你在anaconda安了只需:

否则依次运行以下命令:

最后一句就是安装pytorch1.2.0,切记版本不可错,错了第二步编译会报错。
然后就是跑代码。

跑代码

第一步:进入csrc文件并且编译

不报错就是成功,错误原因都是版本,改一下就好了,接着退回主文件夹:

然后复制官网命令:

记得要分开

无错误就是成功。这里我补充一点,有的小伙伴私信我说遇到了:
错误,这个如果你是在centons,且gcc8
的环境下一般不会报错,但是如果仍然报错了,解决办法参照这个网址:https://github.com/GXYM/DRRG/issues/4,可以解决虽然不是特别完美的办法,但是可以保证顺利运行。
接下来退出两级:

然后训练:进入tool:

运行:

运行,一般会出出两个错误:一个是网络错误再执行以下上面命令就行了。
另外一个错误是没有对应文件夹:只要在代码主文件夹下建立相应文件夹就能解决。
然后就等训练了。

最后

写到这里,说明你的环境配置好了,接下来就等他训练完成吧,然后得测试只要用

退回主文件夹然后复制官网命令就好了,这里我再补充一点,就是官网命令没有指定gpu可能会报错,所以你的命令应该是:

意思是用名字为0(就是第一块)的gpu跑。
这就是所有的运行方法了,本人亲测,希望能帮到您。如果有效记得点赞哦。另外如果有神魔问题可以私信我或者评论,我会看到以后解答,因为本人最近小论文也比较忙可能回复不及时还请见谅。

来源:葛葛葛立鹏啊

声明:本站部分文章及图片转载于互联网,内容版权归原作者所有,如本站任何资料有侵权请您尽早请联系jinwei@zod.com.cn进行处理,非常感谢!

上一篇 2020年10月6日
下一篇 2020年10月6日

相关推荐