linux安装cuda

2024-09-07

下载

  wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run


安装

  sudo sh cuda_12.1.0_530.30.02_linux.run

  翻页阅读用户协议,最后输入accept

  根据本机是否已经安装勾选Driver,其余默认。


配置环境变量

  vim ~/.bashrc


  在末尾添加如下内容(依据NVIDIA官方文档所述)

  export CUDA_HOME=/usr/local/cuda-12.1

  export PATH=$CUDA_HOME/bin:$PATH

  export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH


生效环境变量

source ~/.bashrc