1.参考两个博客。
https://blog.youkuaiyun.com/heiheiya/article/details/81454737
https://blog.youkuaiyun.com/qq_25241325/article/details/90753830
2.记录一下我的配置
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-9.0
export PATH=/usr/local/cuda-9.0/bin:$PATH