2. Installation¶
UCTB is based on several well-known deep learning frameworks, including PyTorch, TensorFlow, and MXNet. If you have an Nvidia GPU installed on your computer, we highly recommend you install the GPU version of these frameworks.
UCTB toolbox may not work successfully with the upgrade of some packages. We thus encourage you to use the specific version of packages to avoid unseen errors. To avoid potential conflict, we highly recommend you install UCTB vis Anaconda.
2.1. Install via Anaconda¶
Step 1: Install Anaconda
You can skip to step 2 if you already installed Anaconda.
To install Anaconda, please refer to this page https://www.anaconda.com/download.
Step 2: create UCTB environment
Create the UCTB environment by the following command. You may need the environment.yaml file.
conda env create -f environment.yaml
Then activate the UCTB environment and start to use it. 🎉🎉🎉
conda activate UCTB
2.2. Check for Success¶
If you successfully install UCTB, you may get the following output after importing UCTB.
(UCTB) XXX:~$ python
Type "help", "copyright", "credits" or "license" for more information.
>>> import UCTB
Using TensorFlow backend.
>>>
2.3. Q & A¶
Q: I fail to install PyTorch, TensorFlow, and MXNet, what is the version number of them?
A: We recommend you install PyTorch==1.1.0, TensorFlow==1.13.1, and MXNet==1.5.0 with cuda version 10.0. We here give the installation command:
# Install PyTorch
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch
# Install TensorFlow
conda install tensorflow-gpu==1.13.1
# Install MXNet
pip install mxnet-cu100==1.5.0
Q: I’m using Windows OS, and my Anaconda reports that it cannot find the PyTorch 1.1.0 packages. How to install it?
A: You could install it by the following command.
pip install torch==1.1.0 torchvision==0.3.0 -f https://download.pytorch.org/whl/cu100/torch_stable.html
Q: I fail to install UCTB via pip. How to install it ?
A: This situation may occur when your OS is Windows. You could install UCTB by its source code. First download UCTB’s source code and your folder may look like this:
XXX/UCTB-master# ls
build_install.py dist environment.yaml __init__.py QuickStarts setup.py
build.py docs Experiments LICENSE README.md UCTB
then build and install UCTB by the following command:
python build_install.py