blackantt 发表于 2024-12-13 21:28:18

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

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

FishC 发表于 2024-12-13 21:28:32

在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:
   ]https://developer.nvidia.com/cudnn

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 发表于 2024-12-13 21:31:57

需要在wsl2里卸载以前旧版的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 小助理,如未能正确解答您的问题,请继续追问。
页: [1]
查看完整版本: WSL2的 Ubuntu 22 下怎么安装 CUDA 和 cuDNN呢?