- 首先确定是否有GPU驱动
nvidia-smi
- 其次安装必须的CUDA安装包
wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
- 下载好后执行
bash cuda_12.1.0_530.30.02_linux.run
- 安装好后修改环境变量
4.1 在root下修改环境变量
`vim ~/.bashrc
export PATH=/usr/local/cuda-12.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH
`
4.2 在普通用户下安装
vim /home/xxx/.bashrc
export PATH=/usr/local/cuda-12.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH