NVLink是NVIDIA的一种高速GPU互连技术,用于连接多个GPU以实现更高的计算性能
首先,确保您的系统已安装了NVIDIA CUDA工具包。如果没有,请访问NVIDIA CUDA工具包下载页面并按照说明进行安装。
安装完成后,重启计算机以使更改生效。
打开终端并运行以下命令以安装NVLink驱动:
sudo apt update
sudo apt install nvidia-driver-<version>-nvlink
将<version>
替换为与您的CUDA工具包匹配的NVIDIA驱动程序版本。例如,如果您安装了CUDA 11.0,则可以使用以下命令:
sudo apt install nvidia-driver-450-nvlink
安装完成后,重启计算机以使更改生效。
要验证NVLink是否正常工作,请运行以下命令:
nvidia-smi nvlink
这将显示有关NVLink连接的信息,包括连接状态、数据传输速率等。
现在,您已经在Ubuntu上安装了NVLink驱动。如果您在安装过程中遇到任何问题,请查阅NVIDIA官方文档以获取更多帮助。