【Linux】安装cuda
一、安装nvidia驱动
# 添加nvidia驱动ppa库
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
# 查找推荐版本
sudo ubuntu-drivers devices
# 安装推荐版本
sudo apt install nvidia-driver-560
# 检验nvidia驱动是否安装
nvidia-smi
二、安装cudatoolkit(cuda)
官网:https://developer.nvidia.com/cuda-toolkit-archive
- 查看nvidia驱动支持的cuda版本
- 按照官网提示去操作
- 配置环境变量
# 如果shell是zsh
sudo vim ~/.zshrc
# 添加如下内容:
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
# 刷新配置文件
source ~/.zshrc
- 验证是否安装成功
nvcc --version