在 luarocks 中安装 cutorch 时无法构建“cutorch”

问题描述


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

luarocks 安装 cutorch

以下是我的错误内容

CMake 错误:此项目中使用了以下变量,但它们被设置为 NOTFOUND。 请设置它们或确保它们在 CMake 文件中正确设置和测试: CUDA_cublas_device_LIBRARY(高级) 由目录 /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”。

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

附言 我已经解决了无法通过 TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" ./install.sh 和 ./update

生成错误生成文件”的问题

解决方法

export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__" && luarocks install cutorch