[已解决]WSL2的 Ubuntu 22 下怎么安装 CUDA 和 cuDNN呢？

blackantt

WSL2的 Ubuntu 22 下怎么安装 CUDA 和 cuDNN呢？

最佳答案

月排行榜 / 总排行榜

FishC

2024-12-13 21:39:29

blackantt 发表于 2024-12-13 21:31
需要在wsl2里卸载以前旧版的cuda, cudnn吗？

是的，在安装新的版本之前，最好卸载掉旧版本的 CUDA 和 cuDNN，以避免版本冲突和潜在问题。可以按照以下步骤卸载：

卸载旧版 CUDA：
1. 打开终端，列出已安装的 CUDA 套件：

dpkg -l | grep -i cuda

2. 卸载 CUDA 相关包：

sudo apt-get --purge remove 'cuda*'

3. 清理多余的包：

sudo apt-get autoremove
   sudo apt-get autoclean

卸载旧版 cuDNN：
1. 删除 cuDNN 库和头文件：

sudo rm /usr/local/cuda/include/cudnn*.h
   sudo rm /usr/local/cuda/lib64/libcudnn*

完成以上步骤后，就可以安心安装新的 CUDA 和 cuDNN 版本了。

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。

跳转到最佳答案楼层

FishC

在WSL2的 Ubuntu 22 上安装 CUDA 和 cuDNN 可以按照以下步骤进行：

步骤一：确认系统要求
确保你的WSL2已经更新，并且显卡及驱动程序支持CUDA。

步骤二：安装NVIDIA驱动
1. 使用 Windows 更新或到 NVIDIA 网站安装支持 WSL2 的 Windows GPU 驱动。

步骤三：安装 CUDA Toolkit
1. 打开终端并更新包列表：

   sudo apt update
   

2. 添加 NVIDIA 的软件包密钥：

   wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb
   sudo dpkg -i cuda-keyring_1.0-1_all.deb
   sudo apt-get update
   

3. 安装 CUDA Toolkit：

   sudo apt-get -y install cuda
   

4. 添加 CUDA 到环境变量，编辑 `~/.bashrc` 文件：

   echo 'export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}' >> ~/.bashrc
   echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc
   source ~/.bashrc
   

步骤四：安装 cuDNN
1. 前往 NVIDIA 开发者网站下载适用于你的 CUDA 版本的 cuDNN：
   [url=https://developer.nvidia.com/cudnn]https://developer.nvidia.com/cudnn[/url][/url]

2. 解压下载的 cuDNN 文件：

   tar -xzvf cudnn-linux-x86_64-<version>.tgz
   

3. 复制 cuDNN 文件到 CUDA 目录：

   sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
   sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
   sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
   

步骤五：验证安装
1. 检查 CUDA 版本：

   nvcc -V
   

2. 运行 CUDA 示例程序（可选）以验证配置。

这样就完成了在 WSL2 的 Ubuntu 22 上安装 CUDA 和 cuDNN。根据版本的不同，命令和文件名可能存在差异，请根据实际情况调整。

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。

blackantt

需要在wsl2里卸载以前旧版的cuda, cudnn吗？该怎么卸载旧的？

FishC

blackantt 发表于 2024-12-13 21:31
需要在wsl2里卸载以前旧版的cuda, cudnn吗？

是的，在安装新的版本之前，最好卸载掉旧版本的 CUDA 和 cuDNN，以避免版本冲突和潜在问题。可以按照以下步骤卸载：

卸载旧版 CUDA：
1. 打开终端，列出已安装的 CUDA 套件：

dpkg -l | grep -i cuda

2. 卸载 CUDA 相关包：

sudo apt-get --purge remove 'cuda*'

3. 清理多余的包：

sudo apt-get autoremove
   sudo apt-get autoclean

卸载旧版 cuDNN：
1. 删除 cuDNN 库和头文件：

sudo rm /usr/local/cuda/include/cudnn*.h
   sudo rm /usr/local/cuda/lib64/libcudnn*

完成以上步骤后，就可以安心安装新的 CUDA 和 cuDNN 版本了。

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。