这花了我很长时间才弄清楚,所以我决定省去人们的麻烦。
尝试在装有 Amazon Linux(redhat( 的 ec2 实例上安装 torch 失败,并显示以下命令:
git clone https://github.com/torch/distro.git ~/torch --recursive
cd ~/torch; bash install-deps;
./install.sh
它给了我错误:
OpenBLAS Failed to compile
事实证明,您只是运行 install-deps 脚本并忽略失败,然后运行以下代码行:
export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__"
并像往常一样按照 install.sh 脚本进行操作:
./install.sh
来源在这里