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