Anaconda 使用conda生成caffe2失败

我用anaconda在页面后面构建了caffe2

在只有一个titanx的服务器中，有cudnn7和cuda9，但没有nccl，因此我从nvidia下载nccl2并将其解压缩到path/to/local/nccl2，然后编辑第42行中的./pytorch/conda/integrated/build.sh为：export nccl_ROOT_DIR=path/to/local/nccl2

然后我需要将caffe2与python2一起使用，因此我在./pytorch/scripts/build_anaconda.sh中添加了conda_args+=-python2.7来使用python2.7

构建成功了，但是当我从caffe2.python导入核心运行python2 test.py时

它告诉我：

警告：root:此caffe2 python运行不支持GPU。将在仅CPU模式下运行

警告：root:调试消息：没有名为caffe2_pybind11_state_hip的模块

分段故障堆芯

我的问题是:

a。为什么conda不支持gpu

b。如果我使用单个gpu，是否需要nccl进行构建

c。如何修复名为caffe2_pybind11_state_hip的模块

PyTorch或Caffe2:Caffe2

如何安装PyTorch conda，pip，来源：conda

从源代码编译时使用的Build命令：./scripts/Build\u anaconda.sh-本地安装-cuda 9.0-cudnn 7

OS:ubuntu16

PyTorch版本：

Python版本：2.7

CUDA/cuDNN版本：9.1/7

GPU型号和配置：

从源代码5.4.0编译时的GCC版本

CMake版本：未安装

任何其他相关库的版本：

---

多谢各位

首先获取CUDA并安装它：

sudo apt-get update && sudo apt-get install wget -y --no-install-recommends
wget "http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb"
sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

现在从源代码开始安装，在环境中执行：

完整的\u CAFFE2=1 python setup.py安装

---

您可以在这里找到更多信息：

请按照以下步骤操作，这对我很有用

ubuntu@test:~$ cd $HOME

ubuntu@test:~$ conda create -n caffe2

ubuntu@test:~$ source activate caffe2

(caffe2) ubuntu@test:~$ git clone --recursive https://github.com/pytorch/pytorch.git && cd pytorch


(caffe2) ubuntu@test:~/pytorch$ git submodule update --init


(caffe2) ubuntu@test:~/pytorch$ CONDA_INSTALL_LOCALLY=1 ./scripts/build_anaconda.sh --cuda 8.0 --cudnn 7 -DUSE_CUDA=ON -DUSE_NCCL=ON