今天想跑一个光流估计的代码,配置环境时从dockerhub直接拉了一个干净的cuda8.0镜像,配到最后才发现有一个脚本执行的时候提示:

nvcc:command not found

明明是从nvidia的官方dockerhub上拉的镜像,难道cuda没有装?

开始怀疑是环境变量没有配置好,于是检查~/.bashrc:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64
export PATH=$PATH:/usr/local/cuda-8.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-8.0

没有问题。

最后经过排查,找到了问题所在:

我最开始拉的镜像是8.0-cudnn7-runtime-ubuntu16.04,但是,ncvv作为cuda的编译器,在runtime的镜像中是不提供的。

所以从一开始就错了,一天的工作白干(哭)

真正应该拉的镜像是8.0-cudnn7-devel-ubuntu16.04,即:

docker pull nvidia/cuda:8.0-cudnn7-devel-ubuntu16.04

唉这周的任务又要延期了,算刚参加工作,经验不足吧。

不过在查询原因的时候发现了一篇不错的文章,分享给大家:

显卡,显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn到底是什么?

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐