CUDA安装教程+显卡驱动安装
新的ubuntu系统,需要重新安装显卡驱动,以及cuda,记录以下坑点:
先安装显卡
nvidia-smi
输入后出现
首先知道自己的GPU型号,如RTX 4090
这里下载:https://www.nvidia.com/Download/index.aspx
搜索之后,下载即可,然后进行安装:
sh .run文件存放的路径
安装成功后键入 nvidia-smi 命令,看是否安装成功。
期间可能会出现没有安装gcc或者make的情况,通过下面命令安装:
sudo apt-get install gcc
gcc --version
sudo apt-get update
sudo apt-get install make
再安装CUDA
- 选择自己需要的CUDA版本,这里选择的是11.8
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
下载完直接安装即可
sudo sh cuda_11.8.0_520.61.05_linux.run
因为之前已经装了显卡,所以记得把第一行勾选去掉,enter即可
然后install,enter
- 添加环境变量
vim ~/.bashrc #编辑文件
#将下面两行插入到文件末尾
export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
source ~/.bashrc #更新文件
- 搞定
nvcc -V