Mamba/PyTorch WSL环境配置

Mamba的配置要求

• Linux
• NVIDIA GPU
• PyTorch 1.12+
• CUDA 11.6+
• https://github.com/state-spaces/mamba

个人版本：

• 通过Windows中的WSL来实现linux环境
• CUDA 12.4
• PyTorch 2.5.1
• Python 3.9+

1、下载并配置WSL

• 微软应用商店搜索wsl选择合适的ubuntu版本进行下载
• 如果遇到报错信息，在windows的cmd中使用wsl --update命令，再尝试打开wsl进行用户名和密码的相关配置。

2、使用VSCode远程连接到WSL

• 在vscode中搜索wsl或wsl remote拓展进行安装，安装完后会出现远程资源管理器选项，在其中选择wsl进行连接和配置。
• 新建终端，就可以在linux环境下运行终端了，并且左下角会显示wsl的连接情况和ubuntu的版本。

3、安装CUDA（内含NVCC, 无需额外下载）

• 在linux终端使用nvidia-smi命令查看cuda和显卡信息，如果没有，则安装cuda。
• 去cuda的官网，选择合适的配置进行下载，CUDA12.4下载链接
  
• 安装完成后通过nvidia-smi和nvcc --version来查看是否安装成功以及安装版本和显卡信息。
• 踩坑记录：
  • 当你下载好文件，并运行文件时，会跳出一个让你同意的协议，但是vscode的终端中可能只有协议，而没有同意选项，你需要在windows的终端中进入wsl，然后再重新安装下载的文件，这时就可以输入accept来同意了。
  • 安装好后，当运行nvcc --version时，提示你找不到nvcc的命令，千万不要按照他的提示来，而是要更改下环境变量，请运行以下命令
    • export PATH=/usr/local/cuda-12.4/bin${PATH:+:${PATH}}
    • 这个12.4是你安装的cuda的版本号，具体的文件名称可以去linux文件管理器中去看
  • 如果不慎运行了他提示的命令怎么办？
    • sudo apt-get --purge remove "*cuda*" "*cublas*" "*cufft*" "*cufile*" "*curand*" \ "*cusolver*" "*cusparse*" "*gds-tools*" "*npp*" "*nvjpeg*" "nsight*" "*nvvm*"
    • sudo apt-get autoremove --purge -V
    • 上述为卸载命令，然后再重新安装即可。
  • 具体文档可以查阅适用于 Linux 的 NVIDIA CUDA 安装指南

4、安装Python3

• 使用python3 --version查看python3的版本，如果没有则进行安装
• sudo apt-get update
• sudo apt-get install python3-pip

5、安装PyTorch

• pytorch2.5.1下载链接
  

6、在Linux终端中运行以下代码，检查是否全部安装成功

• python3
• import torch
• torch.cuda.is_available()
• //如果为true就是成功了
• exit()退出python

7、Mamba启动！

• git clone https://github.com/state-spaces/mamba.git
• pip install .

原文地址：https://blog.csdn.net/wechatonly/article/details/144009356

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！