0

我在页面之后使用 anaconda 构建 caffe2。

在单titanx的服务器上，有cudnn7和cuda9但没有nccl，所以我从nvidia下载nccl2并解压到path/to/local/nccl2，然后编辑./pytorch/conda/integrated/build第 42 行中的 .sh 为：“export NCCL_ROOT_DIR=path/to/local/nccl2”。

然后我需要将caffe2与python2一起使用，所以我在./pytorch/scripts/build_anaconda.sh中添加了“conda_args+=(”--python 2.7")”来使用python2.7。

构建成功，但是当我从 caffe2.python import core 运行 python2 test.py

它告诉我：

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

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

分段错误（核心转储）

我的问题是：

一个。为什么conda不支持gpu？

湾。如果我使用单个 gpu，构建是否需要 nccl？

C。如何修复没有名为 caffe2_pybind11_state_hip 的模块

PyTorch 或 Caffe2：caffe2

你是如何安装 PyTorch 的（conda、pip、source）：conda

您使用的构建命令（如果从源代码编译）：./scripts/build_anaconda.sh --install-locally --cuda 9.0 --cudnn 7

操作系统：ubuntu16

PyTorch 版本：

Python版本：2.7

CUDA/cuDNN 版本：9.1/7

GPU型号和配置：??

GCC 版本（如果从源代码编译）：5.4.0

CMake 版本：未安装

任何其他相关库的版本：

非常感谢！

4

2 回答 2

0

首先获取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

现在继续从源安装（在环境中进行）：

FULL_CAFFE2=1 python setup.py install

您可以在此处找到更多信息：https ://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=compile#install-with-gpu-support

于 2018-09-02T14:02:38.270 回答
-1

请按照以下程序对我有用

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
于 2018-09-19T05:13:05.680 回答