CUDAでエラー nvcc: command not found

Ubuntu 18.04

CUDA 10.0

物体検出YOLOのコンパイル中に表題のエラー

$ nvcc

Command 'nvcc' not found, but can be installed with:

sudo apt install nvidia-cuda-toolkit

しかし、安易にnvidia-cuda-toolkitを入れないよう注意する

KerasやTensorflowを使うために揃えた色々なバージョン、依存関係が崩れ去ってしまうからだ

初心に帰って環境変数の確認をしたところ、単にパスが通っていなかった様子

export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"

nvccが動くか確認

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130

これでOK

参考

Installation Guide Linux :: CUDA Toolkit Documentation
The installation instructions for the CUDA Toolkit on Linux.

コメント

タイトルとURLをコピーしました