4. 开始运行项目¶
运行项目前,需要先安装项目对应依赖环境。
4.2. 依赖安装¶
对于各个任务(image,text,audio),安装的包参见其目录下的requirements.txt
pip install -r requirement.txt
在GPU的方式下,需要根据硬件信息,安装对应的硬件驱动,cuda和cudnn版本。
4.3. 安装torch¶
torch 的安装参照官网,根据用户的环境选择合适的,例如在GPU下, Compute Platform 选择cuda:
CPU下, Compute Platform 选择CPU。
以图像模块 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
=================== =========