File "/anaconda3/envs/lora/lib/python3.10/site-packages/peft/utils/save_and_load.py", line 445, in set_peft_model_state_dict
load_result = model.load_state_dict(peft_model_state_dict, strict=False, assign=True)
TypeError: Module.load_state_dict() got an unexpected keyword argument 'assign'
搜了一大圈之后才搞清楚原来assign关键字是在 Pytorch 2.1.x 新增的，而我自己的版本是 Pytorch 2.0.x，所以并没有这个参数…
解决方法是把peft版本降低到peft==0.12.0，这个版本是支持Pytorch 2.0.x的。 而从peft==0.13.0开始就引入了上述不兼容的代码。
conda创建虚拟环境后，从Pytorch官网下载需要的版本。 其他的包缺啥装啥就行。conda和pip记得换源，不然可能会很慢。
conda 换源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
pip 换源
conda activate <env_name>
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
安装MuJoCo210
—
一般安装的都是MuJoCo210版本，首先去Github上下载对应的压缩包（例如mujoco210-linux-x86_64.tar.gz），并且解压（或者移动）到~/.mujoco/mujoco210文件夹。
解压命令
tar -zxvf <filename.tar.gz>
设置环境变量
然后在~.bashrc中把它添加到LD_LIBRARY_PATH里，如果已有别的版本的mujoco记得屏蔽掉：
export LD_LIBRARY_PATH=/home/wangxun/.mujoco/mujoco210/bin:$LD_LIBRARY_PATH
最后安装2.1.x的mujoco-py即可，例如：mujoco-py==2.1.2.14
首先把D4RL和mjrl都从Github上克隆下来
克隆命令
git clone git@github.com:Farama-Foundation/D4RL.git
git clone git@github.com:aravindr93/mjrl.git
然后进入虚拟环境，进入mjrl文件夹，用pip安装mjrl
安装命令
cd /your_path_to_mjrl
pip install -e .
然后进入D4RL文件夹，注释掉setup.py中的mjrl链接
@@ -68,6 +68,6 @@
"termcolor", # adept_envs dependency
"click", # adept_envs dependency
"dm_control>=1.0.3",
- "mjrl @ git+https://github.com/aravindr93/mjrl@master#egg=mjrl",
+ #"mjrl @ git+https://github.com/aravindr93/mjrl@master#egg=mjrl",
],
)
然后用同样的方法安装D4RL
cd /your_path_to_D4RL
pip install -e .
完成后测试是否能够成功导入d4rl包。
确认点1
/usr/lib/nvidia是否添加到环境变量，如果没有，执行下面的命令：
export LD_LIBRARY_PATH=/usr/lib/nvidia:$LD_LIBRARY_PATH
确认点2
Cython版本是否兼容，如果不兼容，安装以下版本：
pip install Cython==3.0.0a10
确认点3
是否安装X11开发工具包，如果未安装，通过以下命令安装：
sudo apt-get update
sudo apt-get install libx11-dev
确认点4
是否安装GL/glew开发工具包，如果未安装，通过以下命令安装：
sudo apt-get update
sudo apt-get install libglew-dev