在 MacOS 上，PyTorch 未启用和访问 GPU 驱动程序和 CUDA

Question

PyTorch 未启用和访问 GPU 驱动程序和 CUDA。 torch.cuda.is_available() 返回false

我正在使用 macOS Mojave 10.14.6 我已经安装了 pytorch 的 Cuda 10.0 版本。 我在https://pytorch.org/get-started/locally/ 上尝试了验证，并构造了一个随机初始化的张量，效果很好。

但是当我尝试时

import torch
torch.cuda.is_availalbe()

它返回 false。

因此，我按照 Pytorch 上的说明安装了 Anaconda 和 Cuda。

然后试试这个：

git clone --recursive https://github.com/pytorch/pytorch cd pytorch MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py install

在终端，我得到了

fatal error: 'string.h' file not found
#include_next <string.h>

我在 stackoverflow 上搜索并想出了这个。 Build Pytorch from source 。所以我尝试了：

$ find /Library/Developer/CommandLineTools/usr -type f -name string.h

返回/Library/Developer/CommandLineTools/usr/include/c++/v1/string.h

这不是说我已经有string.h了吗？

我该如何解决这个问题？

Answer 1

您是从 conda 环境安装的吗？根据github，这应该可以工作： - 创建一个 conda 环境 - conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing（安装一些要求） 然后这个（我假设你已经完成了）：

git clone --recursive https://github.com/pytorch/pytorch
cd pytorch
# if you are updating an existing checkout
git submodule sync
git submodule update --init --recursive

最后设置 conda 变量并安装：

export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py install

git 报告行为 here 存在问题，建议添加如下内容： MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ NO_CUDA=1 python setup.py install

检查NO_CUDA，这个问题在论坛HERE也提到过，它似乎是操作系统和驱动程序版本引起的问题。如果是这种情况，我建议使用Nvidia Docker（希望它支持mac）和来自https://ngc.nvidia.com/catalog/landing 的pytorch 容器 它也应该失败，你最好的选择是在没有 CUDA 支持的情况下安装。