安装NVIDIA-CUDA组件(含GPU驱动)
操作步骤
- 使用PuTTY工具,以root用户登录服务器。
- 执行以下命令确认nouveau驱动已禁用(回显为空)。
lsmod |grep nouveau
- 执行以下命令安装NVIDIA-CUDA组件。
wget https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda_11.4.1_470.57.02_linux_sbsa.run sh cuda_11.4.1_470.57.02_linux_sbsa.run
根据提示安装所有组件。
- 执行以下命令配置环境变量。
export PATH=/usr/local/cuda-11.4/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH
- 执行以下命令查看nvcc版本。
nvcc -V
父主题: 配置编译环境