在 Amazon Linux 上安装 Torch



这花了我很长时间才弄清楚,所以我决定省去人们的麻烦。

尝试在装有 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

来源在这里

最新更新