准备工作
确定你有支持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