此部分坑多
参考了 cuda安装与可能遇到的问题
https://www.jianshu.com/p/f71bbf2ecde3
1)安装完显卡驱动和cuda及cudnn后,需要配置好环境变量
2)在~/.bashrc中添加如下
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
使环境变量生效
source ~/.bashrc
3)配置 /etc/ld.so.conf.d/cuda.conf
添加如下内容
/usr/local/cuda/lib64
使库生效
sudo ldconfig