Question

GPU驱动程序和CUDA未启用，PyTorch无法访问。 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 env安装吗？根据github，这应该工作： -创建一个conda env -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与https://ngc.nvidia.com/catalog/landing的pytorch容器一起使用（希望它具有mac支持）它也应该失败，最好的选择是在没有CUDA支持的情况下进行安装。

MacOS上的PyTorch无法启用和访问GPU驱动程序和CUDA

1 个答案: