本文是先安装显卡驱动后进行的操作
查看显卡驱动支持CUDA的最新版本12.0
nvidia-smi
检查gcc版本
gcc -v
查看系统支持的gcc版本
https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
选择对应的安装cuda命令
https://developer.nvidia.com/cuda-toolkit-archive
下载安装文件
wget https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run
sudo sh cuda_12.0.0_525.60.13_linux.run
空格键取消 [ ] Driver前面的X,不安装显卡驱动,然后选择install进行安装
查看目录
cd /usr/local/cuda-12.0/bin/
配置环境变量
# 1.nano ~ .bashrc
进入.bashrc文件后, 移动光标:上下左右方向键 ↑ ↓ ← →
移动到文本最下面末尾添加两行:
export PATH=/usr/local/cuda-12.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.0/lib64:$LD_LIBRARY_PATH
# 2.按ctrl+x
按 Y 保存退出
# 3. 应用生效
source .bashrc