Ubuntu 18.04 安装 CUDA

背景

最近在折腾一些github上的代码。由于需要pytorch上需要编译C++，之前通过conda配置的 pytorch+ cudatoolkit是不够的，so需要在ubuntu上安装下cuda。折腾太久，记录一篇。

步骤

1. 挑选合适的版本 。runfile模式，这时官网会提供两行代码，保存好。类似这种

wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run
sudo sh cuda_11.3.0_465.19.01_linux.run

2. 在本地新建一个文件夹，用来装下载的位置。在文件夹内打开终端。直接运行第一行代码，稍等片刻下载。然后运行第二段代码，稍等一会。
3. 这时我这边直接提示已存在驱动，直接选择最下面的continue。然后参考一篇文章https://littlede.blog.csdn.net/article/details/109407098，取消驱动前面的X，安装耐心等待完成。这里步骤遇到不认识的百度解决。
4. 接下来就加路径。（下面的数字 11.3记得改成相应版本）

export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.13/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

5. 验证

nvcc --version

参考文章

[1]https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&=Ubuntu&target_version=18.04&target_type=runfile_local
[2] https://developer.nvidia.com/zh-cn/cuda-gpus
[3] https://littlede.blog.csdn.net/article/details/109407098