forked from xiaofengShi/CHINESE-OCR
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.sh
19 lines (15 loc) · 982 Bytes
/
setup.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# conda create -n chinese-ocr python=2.7 pip scipy numpy PIL jupyter##运用conda 创建python环境
# source activate chinese-ocr
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
pip install easydict -i https://pypi.tuna.tsinghua.edu.cn/simple/ ##选择国内源,速度更快
pip install keras==2.0.8 -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install Cython opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install -U pillow -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install h5py lmdb mahotas -i https://pypi.tuna.tsinghua.edu.cn/simple/
conda install pytorch=0.1.12 cuda90 torchvision -c soumith
pip install futures==3.1.1 -i https://pypi.tuna.tsinghua.edu.cn/simple/
# conda install tensorflow=1.3 tensorflow-gpu=1.3 ##解决cuda报错相关问题
cd ./ctpn/lib/utils
sh make.sh