【代码调试】使用华为云训练自己的代码

麒麒哈尔

于 2020-02-16 11:08:31 发布

阅读量1.8k

点赞数 2

CC 4.0 BY-SA版权

分类专栏：杂七杂八

本文链接：https://blog.youkuaiyun.com/wqwqqwqw1231/article/details/104339524

杂七杂八专栏收录该内容

4 篇文章

订阅专栏

本文解决在华为云环境中遇到的libNVVM找不到的问题，通过在Python脚本中直接添加路径来修正，避免了重复安装的麻烦。适用于在华为云平台上进行深度学习模型训练的场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

libNVVM cannot be found

本人在华为云上调了很多代码，如果有什么问题，可以评论留言，看看能不能解决。

libNVVM cannot be found

错误提示libNVVM cannot be found. Do conda install cudatoolkit。这个问题的来源是路径问题，并不是nvvm没有安装。nvvm在安装cuda的时候就安装了，解决办法可在参考这篇博客。但这个方法并不完全适用，华为云的“训练作业”中的“启动代码”是python脚本，并不是控制台。而bashrc文件是新开一个进程就会加载一遍。所以这里我们选择的方式是，在python脚本中，直接加入这两个路径：

import os
os.system('export NUMBAPRO_NVVM=/usr/local/cuda-9.0/nvvm/lib64/libnvvm.so && export NUMBAPRO_LIBDEVICE=/usr/local/cuda-9.0/nvvm/libdevice/ && echo $NUMBAPRO_NVVM && echo $NUMBAPRO_LIBDEVICE && cd VoteNet_KITTI && python train.py')

在一个指令中，先后加入NUMBAPRO_NVVM和NUMBAPRO_LIBDEVICE的变量，然后确认，然后运行训练脚本。