4. 开始运行项目

运行项目前,需要先安装项目对应依赖环境。

4.1. 基础Python环境

python >= 3.6.5

其中,本项目在 Python 3.6.9 下进行过较为完整的测试。

4.2. 依赖安装

对于各个任务(image,text,audio),安装的包参见其目录下的requirements.txt

pip install -r requirement.txt

在GPU的方式下,需要根据硬件信息,安装对应的硬件驱动,cuda和cudnn版本。

4.3. 安装torch

torch 的安装参照官网,根据用户的环境选择合适的,例如在GPU下, Compute Platform 选择cuda:

image2

CPU下, Compute Platform 选择CPU。

image3

以图像模块 image 为例,安装完成后的Python安装包环境如下:

===================  =========
Package              Version
===================  =========
torch                1.10.1
torchaudio           0.10.1
torchvision          0.11.2
cloudpickle          1.6.0
cycler               0.10.0
dask                 2.27.0
decorator            4.4.2
future               0.18.2
imageio              2.9.0
kiwisolver           1.2.0
matplotlib           3.3.0
networkx             2.5
numpy                1.19.5
opencv-python        3.4.18.65
pandas               0.23.4
Pillow               7.2.0
pyparsing            2.4.7
python-dateutil      2.8.1
pytz                 2020.1
PyWavelets           1.1.1
PyYAML               5.3.1
scikit-image         0.17.2
scipy                1.5.2
six                  1.15.0
tifffile             2020.9.3
toolz                0.10.0
Wand                 0.5.2
tqdm                 4.61.1
===================  =========