Cuda in docker: nvcc command not found解决方法

今天想跑一个光流估计的代码,配置环境时从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到底是什么?

你可能感兴趣的:(Linux和ROS)