linux集群部署深度学习平台Pytorch

Posted 大雨海棠

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了linux集群部署深度学习平台Pytorch相关的知识,希望对你有一定的参考价值。

1. 先安装Python, 从官网 https://www.python.org/getit/ 下载Python 安装包,这里下载的 Python-3.6.0.tgz

tar -xvf Python-3.6.0.tgz
cd Python-3.6.0
./configure --prefix=/share/home/kuangx/opt/python3.6  --enable-shared
# prefix 参数指定要将 python 安装到哪个目录, 我的安装目录是: /share/home/kuangx/opt/python3.6 # enable-shared 是必须要写的,否则后面会遇到错误提示:Command failed with rc=65536
make
make install

将python3.6安装路径加入的环境变量中,打开bashrc文件

export PATH=$PATH:/share/home/kuangx/opt/python3.6/bin

安装完成后,在终端键入

python3

若出现python相关版本信息,说明安装成功。我在安装过程中出现了一个库找不到,将/share/home/kuangx/opt/python3.6/lib 加入环境变量LD_LIBRARY_PATH中,具体如下

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:share/home/kuangx/opt/python3.6/lib

这样问题就解决了。

2. 安装setuptool(为下面安装pip服务)

wget --no-check-certificate https://pypi.python.org/packages/source/s/setuptools/setuptools-19.6.tar.gz#md5=c607dd118eae682c44ed146367a17e26
tar -zxvf setuptools-19.6.tar.gz
cd setuptools-19.6
python3 setup.py build
python3 setup.py install

3.安装pip

wget --no-check-certificate https://pypi.python.org/packages/source/p/pip/pip-8.0.2.tar.gz#md5=3a73c4188f8dbad6a1e6f6d44d117eeb
tar -zxvf pip-8.0.2.tar.gz
cd pip-8.0.2
python3 setup.py build
python3 setup.py install

安装完pip之后,通过如下命令更新pip版本

python3 -m ensurepip
python3 -m pip install --upgrade pip setuptools wheel

4.通过pip安装第三方库

pip3 install numpy
pip3 install sklearn
pip3 install torch

终端输入:

  python3

依次输入 

import numpy
import sklearn
import torch

若不报错,这说明安装成功。

 

以上是关于linux集群部署深度学习平台Pytorch的主要内容,如果未能解决你的问题,请参考以下文章

TVM:成为深度学习领域的“Linux”

PyTorch 深度学习框架:优雅而简洁的代码实现

youcans的深度学习 02PyTorch CPU版本安装与环境配置

youcans的深度学习 02PyTorch CPU版本安装与环境配置

实现Linux服务器配置深度学习环境并跑代码完整步骤

将深度学习模型部署为exe需要哪些工具