安装神经结构化学习

有几种方法可以设置您的环境，以便在 TensorFlow 中使用神经结构化学习 (NSL)

  • 学习和使用 NSL 的最简单方法不需要安装：使用 Google Colaboratory 直接在浏览器中运行 NSL 教程。
  • 要在本地机器上使用 NSL，请使用 Python 的 pip 包管理器安装 NSL 包。
  • 如果您有独特的机器配置，请从源代码 构建 NSL。

使用 pip 安装神经结构化学习

1. 安装 Python 开发环境。

在 Ubuntu 上

sudo apt update
sudo apt install python3-dev python3-pip  # Python 3
sudo pip3 install --upgrade virtualenv  # system-wide install

在 macOS 上

/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
export PATH="/usr/local/bin:/usr/local/sbin:$PATH"
brew update
brew install python  # Python 3
sudo pip3 install --upgrade virtualenv  # system-wide install

2. 创建虚拟环境。

virtualenv --python python3 "./venv"
source "./venv/bin/activate"
pip install --upgrade pip

3. 安装 TensorFlow

CPU 支持

pip install 'tensorflow>=1.15.0'

GPU 支持

pip install 'tensorflow-gpu>=1.15.0'

4. 安装神经结构化学习 pip 包。

pip install --upgrade neural_structured_learning

5. （可选）测试神经结构化学习。

python -c "import neural_structured_learning as nsl"

构建神经结构化学习 pip 包

1. 安装 Python 开发环境。

在 Ubuntu 上

sudo apt update
sudo apt install python3-dev python3-pip  # Python 3
sudo pip3 install --upgrade virtualenv  # system-wide install

在 macOS 上

/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
export PATH="/usr/local/bin:/usr/local/sbin:$PATH"
brew update
brew install python  # Python 3
sudo pip3 install --upgrade virtualenv  # system-wide install

2. 安装 Bazel。

安装 Bazel，这是用于编译神经结构化学习的构建工具。

3. 克隆神经结构化学习存储库。

git clone https://github.com/tensorflow/neural-structured-learning.git

4. 创建虚拟环境。

virtualenv --python python3 "./venv"
source "./venv/bin/activate"
pip install --upgrade pip

5. 安装 Tensorflow

请注意，NSL 需要 TensorFlow 1.15 或更高版本。NSL 还支持 TensorFlow 2.0。

CPU 支持

pip install 'tensorflow>=1.15.0'

GPU 支持

pip install 'tensorflow-gpu>=1.15.0'

6. 安装神经结构化学习依赖项。

cd neural-structured-learning
pip install --requirement neural_structured_learning/requirements.txt

7. （可选）对神经结构化学习进行单元测试。

bazel test //neural_structured_learning/...

8. 构建 pip 包。

python setup.py bdist_wheel --universal --dist-dir="./wheel"

9. 安装 pip 包。

pip install --upgrade ./wheel/neural_structured_learning*.whl

10. 测试神经结构化学习。

python -c "import neural_structured_learning as nsl"