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