在 luarocks 安装 cutorch 时构建“cutorch”失败


  • 操作系统版本:Ubuntu18.04
  • CUDA 工具包版本:CUDA 工具包 10.1 更新 2
  • CUDnn 版本:7.6.2.24

luarocks 安装 cutorch

下面是我的错误内容:

CMake 错误:在此项目中使用了以下变量,但它们被设置为 NOTFOUND。请设置它们或确保在 CMake 文件中已正确设置并测试它们: CUDA_cublas_device_LIBRARY (ADVANCED) 在 /tmp/luarocks_cutorch-scm-1-8576/cutorch/lib/THC 目录中由目标“THC”链接

  • 配置不完整,发生错误!- 请参见“/tmp/luarocks_cutorch-scm-1-8576/cutorch/build/CMakeFiles/CMakeOutput.log”。 请参见“/tmp/luarocks_cutorch-scm-1-8576/cutorch/build/CMakeFiles/CMakeError.log”。

错误:构建错误:构建失败。

P.S. 我通过 TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" ./install.sh 和 ./update 解决了无法生成“Error generating file”的问题。

点赞
用户8846505
用户8846505
export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" && luarocks install cutorch

将上述命令翻译成中文并保留原本的markdown格式:

设置Torch的NVCC标志为“-D__CUDA_NO_HALF_OPERATORS__”,然后安装cutorch。
2021-07-11 12:29:51