1.先裝驅動,以為安裝CUDA時安裝最新驅動導致CUDA用不了
sudo apt-get install nvidia-396
2.參考這,安裝好CUDA 9.2
https://developer.nvidia.com/cuda-92-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal
https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#runfile-advanced
sudo apt-get install cuda-toolkit-9-2
sudo apt-get install cuda-libraries-9-2
然后測試cuda是否安裝成功
https://docs.nvidia.com/cuda/archive/9.2/cuda-quick-start-guide/index.html#ubuntu-x86_64
或者直接下載https://developer.nvidia.com/cuda-92-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal
這個文件用sh安裝,繼續選別更新驅動
3.參考這安裝好cudnn
https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#installlinux-deb