在WSL2-Linux中安装NVIDIA CUDA

如果你是在Windows子系统WSL2-Linux中使用Ubuntu,并且想要安装NVIDIA CUDA以加速计算工作,你来对地方了。本教程将带你一步步完成CUDA的安装和配置,确保一切顺利运行。尽管安装CUDA本身并不复杂,但由于WSL子系统的一些特殊性,需要额外的配置步骤,尤其是解决libcuda.so文件找不到的问题。

前置条件

在开始之前,请确保你已经完成以下前置条件:

  1. 安装了WSL2-Linux子系统。如果你尚未安装,请查阅相关教程。
  2. 打开一个新的命令行窗口,以确保环境是干净的。

步骤1:进入WSL虚拟机

如果你只有一个WSL虚拟机,请执行以下命令进入虚拟机:

wsl

如果你有多个WSL虚拟机,请用以下命令替换“虚拟机名称”:

wsl -d 虚拟机名称

步骤2:切换为root用户

执行以下命令,将用户切换为root用户:

su root

你可能会被要求输入密码,输入后按回车(密码输入过程中不可见)。

步骤3:准备工作

执行以下命令,将当前目录切换到root用户的主目录下:

cd ~

步骤4:下载并安装CUDA

现在,我们将下载和安装CUDA。首先,确定你需要的CUDA版本。不同的软件可能需要特定版本的CUDA,所以请确保你选择正确的版本。

  1. 访问NVIDIA官方网站上的CUDA下载页面。
  2. 找到你想要的CUDA版本,例如CUDA 11.8,并点击该版本下的链接。
  3. 在版本页面中,找到适用于你的系统的选项:
    • Operating System(系统平台):选择Linux。
    • Architecture(系统架构):选择X86_64。
    • Distribution(系统版本):选择WSL-Ubuntu。
    • Version(版本):通常只有一个选项,不用担心。
    • Installer Type(安装方式):选择deb(local)以下载并完整安装CUDA。

生成的代码类似于以下内容:

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-wsl-ubuntu-11-8-local_11.8.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-11-8-local_11.8.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

请注意,如果你选择的是不同版本的CUDA,生成的代码将不同。请根据你的选择修改代码。

步骤5:配置环境变量

安装完CUDA后,我们需要配置环境变量,以便系统可以正确访问CUDA。

首先,创建一个环境变量以引用CUDA的版本。使用以下命令:

CUDA_VERSION="cuda-11.8"  # 如果你安装的是不同版本的CUDA,请将11.8修改为你的实际版本

然后,执行以下命令,将CUDA的路径添加到环境变量中:

echo "export PATH=\"/usr/local/$CUDA_VERSION/bin:\$PATH\"" | sudo tee -a /etc/profile.d/somefile.sh

由于WSL的特殊性,我们需要使用软链接来解决libcuda.so无法找到的问题。执行以下命令:

sudo ln -s /usr/lib/wsl/lib/libcuda.so /usr/local/$CUDA_VERSION/targets/x86_64-linux/lib/libcuda.so

最后,执行以下命令以将链接后的位置添加到LD_LIBRARY_PATH环境变量中:

echo "export LD_LIBRARY_PATH=\"/usr/local/$CUDA_VERSION/targets/x86_64-linux/lib:\$LD_LIBRARY_PATH\"" | sudo tee -a /etc/profile.d/somefile.sh

最后,刷新系统环境以使更改生效:

source /etc/profile.d/somefile.sh

步骤6:验证CUDA

现在,我们已经完成了CUDA的安装和配置。你可以执行以下命令来验证CUDA是否成功安装:

nvcc --version

如果一切设置正确,你应该看到类似以下内容:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:33:58_PDT_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0

如果你想查看CUDA的安装路径,可以执行以下命令:

dirname $(which nvcc)

你将看到类似以下内容:

/usr/local/cuda-11.8/bin

这表示CUDA已成功安装在/usr/local/cuda-11.8/bin目录下。

恭喜!你已经成功在WSL2-Linux中安装和配置了NVIDIA CUDA,现在可以在你的项目中加速计算了。

总结

CUDA的安装虽然简单,但在WSL2子系统中需要特殊的配置。通过本教程,你学会了如何下载、安装和配置NVIDIA CUDA,以便在WSL2-Linux中

加速计算任务。现在,你可以轻松地在CUDA支持的项目中发挥其强大的计算能力。

本文由作者 王大神 原创发布于 大神网的AI博客。

转载请注明作者:王大神

原文出处:在WSL2-Linux中安装NVIDIA CUDA

(0)
打赏 微信扫一扫 微信扫一扫
上一篇 2023年9月13日
下一篇 2023年9月13日