Ubuntu环境下安装CUDA

查看Nvidia显卡驱动信息

nvidia-smi

下载对应版本的CUDA

https://developer.nvidia.com/cuda-toolkit-archive

执行安装文件

sudo sh cuda_版本信息_linux.run

（若driver已经安装过，去掉Driver选项）输入accept

编辑环境变量

vim ~/.bashrc

配置环境变量

export PATH=/usr/local/cuda-版本号/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-版本号/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

启用配置

source ~/.bashrc

查看安装是否成功

nvcc -V