linux虚拟机如何安装cuda

在Linux虚拟机上安装CUDA的步骤如下：

1. 首先，确保你的虚拟机支持GPU并且已经安装了NVIDIA驱动。你可以通过运行以下命令来检查：

lspci | grep -i nvidia

如果输出中包含"NVIDIA"字样，那么你的虚拟机支持GPU。

2. 下载CUDA Toolkit安装包。访问NVIDIA官方网站（https://developer.nvidia.com/cuda-downloads）并选择适合你的操作系统和版本的CUDA Toolkit。例如，对于Ubuntu系统，你可以使用以下命令下载安装包：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda-repo-ubuntu1804-11-4-local_11.4.1-470.57.02-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-11-4-local_11.4.1-470.57.02-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu1804-11-4-local/7fa2af80.pub
sudo apt update
sudo apt install cuda

注意：上述命令中的版本号（例如11.4.1）可能会随着新版本的发布而发生变化，请根据实际情况进行调整。

3. 配置环境变量。在你的`~/.bashrc`文件中添加以下内容：

export PATH=/usr/local/cuda-11.4/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH

注意：上述命令中的路径（例如/usr/local/cuda-11.4）需要根据你的实际安装路径进行修改。

4. 保存文件并运行以下命令使配置生效：

source ~/.bashrc

5. 验证CUDA安装是否成功。运行以下命令：

nvcc --version

如果输出中包含CUDA版本信息，那么恭喜你，CUDA已经成功安装在你的Linux虚拟机上了。<span class="em">1</span><span class="em">2</span><span class="em">3</span><span class="em">4</span>