準備工作
確定你有支持CUDA的NVIDIA GPU
$ lspci | grep -i nvidia
3b:00.0 3D controller: NVIDIA Corporation GP100GL [Tesla P100 PCIe 16GB] (rev a1)
並前往這裏對應位置查看當前GPU是否支持CUDA。
確定要安裝的CUDA Toolkit版本
注意: 如果是用來訓練TensorFlow、Pytorch等深度學習框架,請參考此類深度學習框架GPU版本的編譯環境進行安裝。
版本向下兼容??(不確定,但應安裝較新版本的CUDA)
獲取安裝文件
$ wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
名稱解釋 cuda_10.2.89_440.33.01_linux.run
10.2.89 :CUDA版本
請自行驗證文件的md5碼。
https://blog.csdn.net/qq_40930088/article/details/103389244