setup CUDA path
First of all, check your CUDA:
nvcc -V or nvcc --version
If there is nothing:
export CUDA_HOME=/usr/local/cuda-<version> # Replace <version> with your CUDA version
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
本文作者:Kane,转载请注明原文链接:https://www.cnblogs.com/hackerk/p/18041937