首先确定是否有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下修改环境变量

<code>`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