Paddle paddle多卡训练时配置cudnn的路径问题

93ze6v8z  于 2021-11-30  发布在  Java
关注(0)|答案(7)|浏览(467)

paddle-gpu=2.0.0
cudatoolkit=11.0
采用多卡训练时候,发现paddle好像永远是从系统的/usr/local/cuda这个目录下读取cudnn的信息,由于修改这个目录一般需要管理员权限,因此Paddle是否可以自定义cudatoolkit的根目录或者自定义cudnn的目录?是否有对应的环境变量或者接口设置对应的目录?

mccptt67

mccptt671#

您好,我们已经收到了您的问题,会安排技术人员尽快解答您的问题,请耐心等待。请您再次检查是否提供了清晰的问题描述、复现代码、环境&版本、报错信息等。同时,您也可以通过查看官网API文档常见问题历史IssueAI社区来寻求解答。祝您生活愉快~

Hi! We've received your issue and please be patient to get responded. We will arrange technicians to answer your questions as soon as possible. Please make sure that you have posted enough message to demo your request. You may also check out the APIFAQGithub Issue and AI community to get the answer.Have a nice day!

falq053o

falq053o2#

暂时没有支持自定义cudnn的目录,可以尝试修改安装目录

eqzww0vc

eqzww0vc3#

谢谢回复!
我大概查了一下代码,里面
代码

这里好像提到了修改$LD_LIBRARY_PATH 可以进行配置?

dfuffjeb

dfuffjeb4#

可以尝试修改环境变量的方式,修改路径。

qcbq4gxm

qcbq4gxm5#

在个人用户下的.bashrc中修改LD_LIBRARY_PATH, 但是不起作用,确定修改在系统中是生效的。但是paddle检测到的始终是/usr/local/cuda这个目录下所安装的cudnn版本。
paddle现在是否支持cudatoolkit+cudnn一起打包安装的方式?

还有一个问题,nccl的库目录是否可以用LD_LIBRARY_PATH单独设置呢?

deikduxw

deikduxw6#

支持直接打包安装,或者直接使用docker配置好的环境

rt4zxlrg

rt4zxlrg7#

这个问题太让人困扰了,居然不支持 LD_LIBRARY_PATH, 😞

相关问题