1.检查环境
lspci | grep -i nvidia (查看显卡型号)
uname -m && cat /etc/*release
gcc --version
uname -r
sudo apt-get install linux-headers-$(uname -r) (install与内核一致相关包)
2.安装CUDA、cuDNN、tensorflow, 根据一下表格安装对应的版本,注意一定要对应的版本。
这里wo'xua