如果你是在Windows子系统WSL2-Linux中使用Ubuntu,并且想要安装NVIDIA CUDA以加速计算工作,你来对地方了。本教程将带你一步步完成CUDA的安装和配置,确保一切顺利运行。尽管安装CUDA本身并不复杂,但由于WSL子系统的一些特殊性,需要额外的配置步骤,尤其是解决libcuda.so文件找不到的问题。
前置条件
在开始之前,请确保你已经完成以下前置条件:
- 安装了WSL2-Linux子系统。如果你尚未安装,请查阅相关教程。
- 打开一个新的命令行窗口,以确保环境是干净的。
步骤1:进入WSL虚拟机
如果你只有一个WSL虚拟机,请执行以下命令进入虚拟机:
wsl
如果你有多个WSL虚拟机,请用以下命令替换“虚拟机名称”:
wsl -d 虚拟机名称
步骤2:切换为root用户
执行以下命令,将用户切换为root用户:
su root
你可能会被要求输入密码,输入后按回车(密码输入过程中不可见)。
步骤3:准备工作
执行以下命令,将当前目录切换到root用户的主目录下:
cd ~
步骤4:下载并安装CUDA
现在,我们将下载和安装CUDA。首先,确定你需要的CUDA版本。不同的软件可能需要特定版本的CUDA,所以请确保你选择正确的版本。
- 访问NVIDIA官方网站上的CUDA下载页面。
- 找到你想要的CUDA版本,例如CUDA 11.8,并点击该版本下的链接。
- 在版本页面中,找到适用于你的系统的选项:
- 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支持的项目中发挥其强大的计算能力。