基于Python语言的工程文件,有时候需要对相关库进行封装,以方便在别的电脑上直接运行。最近对实验室基于Pytorch框架和深度学习的工程文件进行了封装,特此记录一下:
1.工程文件依赖的库:
封装之前得弄清楚自己的python工程文件具体依赖哪些库。本次项目我们只依赖于自己在Anaconda下创建的Pytorch环境,相关的库也是集中在这个环境下。
2.拷贝Anaconda环境下的库:
将Anaconda目录下envs中的Pytorch文件拷贝至工程文件中。将其放在工程文件的python文件夹下。
3.编写sh运行文件:
创建run.sh文件,需要设置3个相关路径和运行主文件
#!/bin/bash
#指定寻找路径
export PATH=./自己的工程目录/python/Pytorch/bin
#指定python路径
export PYTHONPATH=./自己的工程目录/python/Pytorch/lib/python3.6:./自己的工程目录/python/Pytorch/lib/python3.6/site-packages
#指定依赖库路径
export LD_LIBRARY_PATH=./自己的工程目录/python/Pytorch/lib
python main.py
4.打开终端,直接
sh run.sh