Mamba_0">Mamba的配置要求
Linux
NVIDIA GPU
PyTorch
1.12+CUDA
11.6+https://github.com/state-spaces/mamba
个人版本:
- 通过
Windows
中的WSL
来实现linux
环境 CUDA
12.4PyTorch
2.5.1Python
3.9+
WSL_11">1、下载并配置WSL
- 微软应用商店搜索
wsl
选择合适的ubuntu
版本进行下载 - 在主板
Bios
中开启虚拟化功能,貌似是在高级
->cpu
中 - 在windows功能中开启
Hyper-V
,里面的管理工具和平台都要勾选,以及勾选适用于Linux的Windows子系统
选项。 - 如果遇到报错信息,在
windows
的cmd
中使用wsl --update
命令,再尝试打开wsl
进行用户名和密码的相关配置。 - 在使用apt命令前,更新一下apt,
apt-get update
WSL_17">2、使用VSCode远程连接到WSL
- 在
vscode
中搜索wsl
或wsl remote
拓展进行安装,安装完后会出现远程资源管理器
选项,在其中选择wsl
进行连接和配置。 - 新建终端,就可以在
linux
环境下运行终端了,并且左下角会显示wsl
的连接情况和ubuntu
的版本。
3、安装CUDA(内含NVCC, 无需额外下载)
- 在
linux
终端使用nvidia-smi
命令查看cuda
和显卡信息,如果没有,则安装cuda
。 - 去
cuda
的官网,选择合适的配置进行下载,CUDA12.4下载链接 gcc
相关错误:apt-get update
,sudo apt install gcc
- 安装完成后通过
nvidia-smi
和nvcc --version
来查看是否安装成功以及安装版本和显卡信息。 - 踩坑记录:
- 当你下载好文件,并运行文件时,会跳出一个让你同意的协议,但是
vscode
的终端中可能由于显示高度太短,只有协议,而没有同意选项,你可以把终端的高度拉高或者在windows
的终端中进入wsl
,然后再重新安装下载的文件,这时就可以输入accept
来同意了。 - 安装好后,当运行
nvcc --version
时,提示你找不到nvcc
的命令,千万不要按照他的提示来,而是要更改下环境变量,请运行以下命令export PATH=/usr/local/cuda-12.4/bin${PATH:+:${PATH}}
- 这个
cuda-12.4
是你安装的cuda的版本号,目前最新是cuda-12.6
,具体的文件名称可以去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
5、安装PyTorch
- pytorch2.5.1下载链接
6、在Linux终端中运行以下代码,检查是否全部安装成功
Mamba_50">7、Mamba启动!
git clone https://github.com/state-spaces/mamba.git
cd mamba
pip install .
8、Anaconda安装
- 报错
conda: command not found
:需要安装Anaconda
来解决,去官网填写邮件后下载。 - Anaconda下载
- 在
Wsl
环境下,我们要下载的是Linux
版本 - 将下载好的文件,移动到
linux
磁盘下的home
->username
目录下,然后到对用的目录去运行 sh Miniconda3-latest-Linux-x86_64.sh
,sh
后面跟的是对应的文件名
Mamba_60">9、U-Mamba启动!
- 项目地址:
https://github.com/bowang-lab/U-Mamba
- 创建虚拟环境:
conda create -n umamba python=3.10 -y
- 激活环境:
conda activate umamba
- 安装
causal-conv1d
:pip install causal-conv1d
,这步可能会出错,wget
命令也会报404 Not Found
,替代办法是直接在github
仓库直接下载这个文件,然后放到linux
磁盘下的home
->username
目录下,然后到对应的目录去运行pip install 对应下载的文件名
。causal-conv1d仓库链接 - 安装
Mamba
:其实我们在第7
条中已经安装mamba
了,官方文档中的pip install mamba-ssm --no-cache-dir
会报404错误。 - 验证是否安装成功:
python3
import torch
import mamba_ssm
- 没报错就是安装上了
exit()
- 下载
U-Mamba
代码:git clone https://github.com/bowang-lab/U-Mamba
- 安装依赖:
pip install -e .
- 将数据集按照格式要求放好,就可以按照项目地址的命令开始训练了
10、连接github太慢,超时,443错误
- 方法很多,但不一定有用,比如设置代理
git config --global http.proxy
git config --global https.proxy
11、pip 太慢,超时(换清华源)
pip install pip -U
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
12、npm太慢,超时(换淘宝源)
npm config set registry https://registry.npmmirror.com