在Gym训练自定义环境调用learn方法遇PyTorch LAPACK相关RuntimeError求助

问题原因 #

你当前使用的PyTorch版本在编译时未包含LAPACK依赖，而torch.geqrf这个线性代数操作必须依赖LAPACK库才能运行，因此调用learn方法时触发了错误。

具体解决办法 #

• 重新安装带LAPACK支持的PyTorch
  卸载现有版本后，安装官方预编译的、默认包含LAPACK的PyTorch包：

  1. 先卸载旧版本：

     pip uninstall torch -y
     

  2. 根据你的系统环境选择对应安装命令，比如CPU版本：

     pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
     

     要是用GPU，替换成对应CUDA版本的官方安装命令即可。

• 验证修复效果
  安装完成后，可通过以下代码确认LAPACK是否生效：

  import torch
  # 测试geqrf操作
  x = torch.randn(3, 3)
  q, r = torch.geqrf(x)
  print("操作成功，LAPACK已正常启用")
  

• 源码编译PyTorch的特殊处理
  如果你是自行编译PyTorch，需要先在系统安装LAPACK开发库：

  • Ubuntu/Debian：sudo apt-get install liblapack-dev
  • CentOS/RHEL：sudo yum install lapack-devel
    然后编译时添加启用LAPACK的参数：

  python setup.py install --cmake-args="-DUSE_LAPACK=ON"
  

内容的提问来源于stack exchange，提问作者Ad3zp