有多種方式可設定您的環境,以便在 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"