使用AutoDL安装Mamba官方代码

使用AutoDL安装Mamba

租界的云服务器使用环境ubuntu22.04, cuda 11.8, cudnn8.9 python= 3.10 torch=2.10

远程连接验证安装条件

1. 使用Pycharm连接远程的云GPU服务器
   
2. 使用nvidia-smi 和 nvcc -V python conda info-e来验证云主机是否具有安装的条件。

conda创建虚拟环境并安装pytorch

在确保云GPU服务器满足安装条件的基础上，我们使用conda来创建对应的虚拟环境。

conda create -n Mamba python=3.10

执行初始化操作并激活所创建的虚拟环境。

source activate

root@autodl-container-1d0242a1d3-c28f8abb:~# source activate
(base) root@autodl-container-1d0242a1d3-c28f8abb:~# 

conda activate Mamba

首先先安装pytorch

conda install pytorch torchvision cudatoolkit=11.8 -c pytorch

在下载完成之后验证一下GPU版本的torch是否可以使用。

python
import torch
print(torch.cuda.is_available())

不可以使用时结合官网文档重新进行安装使用。

conda remove pytorch torchvision cudatoolkit=11.8 -c pytorch

conda install pytorch== 2.2.1 torchvision== 0.17.1 torchaudio==2.2.1 pytorch-cuda=11.8 -c pytorch -c nvidia

pip install torch== 2.2.0 torchvision== 0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu118 -i https://mirrors.aliyun.com/pypi/simple/

>>> import torch
>>> print(torch.cuda.is_available())
True

Mamba项目克隆和安装

(Mamba) root@autodl-container-1d0242a1d3-c28f8abb:~# mkdir chenruxu
(Mamba) root@autodl-container-1d0242a1d3-c28f8abb:~# ls
autodl-pub  autodl-tmp  chenruxu  miniconda3  tf-logs
(Mamba) root@autodl-container-1d0242a1d3-c28f8abb:~# cd chenruxu/
(Mamba) root@autodl-container-1d0242a1d3-c28f8abb:~/chenruxu# ls

1. pip install causal-conv1d>=1.2.0

在AutoDL的GPU服务器中安装不成功，需要我们手动的进行下载并使用。

https://github.com/Dao-AILab/causal-conv1d/releases

2. git clone https://github.com/state-spaces/mamba.git

3. cd mamba pip install .

pip install . --no-build-isolation -i https://pypi.tuna.tsinghua.edu.cn/simple