EVSSM

配置环境:

git clone https://github.com/lksky-kong/EVSSM.git
cd EVSSM
conda env create -f environment.yml
conda activate torch2.1

首先确认GPU驱动版本是否支持CUDA 12.1

nvidia-smi 

命令行输出:NVIDIA-SMI 535.129.03             Driver Version: 535.129.03   CUDA Version: 12.2

,表明支持CUDA 12.1。

安装好其他所有依赖:

pip install -r requirements_basicsr.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

        运行bash train.sh,报错核心:ERROR: Could not install packages due to an OSError: [Errno 2] No such file or directory: '/home/zkk/miniconda3/envs/torch2.1/lib/python3.10/site-packages/torch/include/ATen/ATen.h'

        这表示:pip 在安装或升级 torch 时,检测到旧的 torch 残留文件夹,但其中部分核心头文件(如 ATen.h)缺失,导致安装中断。也就是说环境里 torch 没完全卸载干净,pip 在覆盖时出错。要彻底清除旧的 PyTorch 文件夹后再装。

清除删除残留的torch:

步骤 1:彻底删除残留的 torch 目录

rm -rf /home/zkk/miniconda3/envs/torch2.1/lib/python3.10/site-packages/torch*

步骤2:确认删除成功

ls /home/zkk/miniconda3/envs/torch2.1/lib/python3.10/site-packages/ | grep torch

命令行输出:functorch(没有输出,说明清理干净)

        说明当前 torch 主包确实已经被清理干净,只剩下 functorch(这是 PyTorch 的一个子包,没问题,可以保留)。

安装对应的Pytorch和CUDA:

        在已经安装CUDA 12.1的系统中安装对应的PyTorch,以确保能够利用GPU进行计算

pip install torch==2.1.0+cu121 torchvision==0.16.0+cu121 torchaudio==2.1.0+cu121 -i https://download.pytorch.org/whl/cu121

命令行显示:Successfully installed torch-2.1.0+cu121 torchaudio-2.1.0+cu121 torchvision-0.16.0+cu121,即成功安装。

pip install git+https://github.com/state-spaces/mamba.git@v2.2.3 --no-build-isolation

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值