компиляция исходного кода pytorch gpu

глубокое обучение

Программно-аппаратная среда

  • ubuntu 18.04 64bit
  • NVidia GTX 1070Ti
  • anaconda with python 3.7
  • CUDA 10.1
  • cuDNN 7.6
  • pytorch 1.8

среда Python и GPU

Я не буду больше говорить об этом, если он не установлен, вы можете перейти по следующей ссылке

Шаг компиляции

Установить базовые зависимости

conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses

из-за использованияgpu, поэтому вам также необходимо установитьLAPACKподдержку, согласноCUDAУстановите соответствующий программный пакет

# Add LAPACK support for the GPU if needed
conda install -c pytorch magma-cuda101  # or [ magma-cuda101 | magma-cuda100 | magma-cuda92 ] depending on your cuda version

Затем вы можете начать клонировать код

git clone --recursive https://github.com/pytorch/pytorch
cd pytorch
# if you are updating an existing checkout
git submodule sync
git submodule update --init --recursive

После завершения подготовки можно приступать к сборке

export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
python setup.py install

CMAKE_PREFIX_PATHНа самом деле этоanacondaКаталог установки, например мой здесь/home/xugaoxiang/anaconda3

pytorch

Сборка занимает много времени, просто терпеливо ждите

контрольная работа

открыть одинterminalпровести тестирование

(base) xugaoxiang@1070Ti:~$ ipython
Python 3.7.6 (default, Jan  8 2020, 19:59:22)
Type 'copyright', 'credits' or 'license' for more information
IPython 7.19.0 -- An enhanced Interactive Python. Type '?' for help.

In [1]: import torch

In [2]: torch.__version__
Out[2]: '1.8.0a0+46d846f'

In [3]: torch.cuda.is_available()
Out[3]: True

In [4]:

инструкцияpytorchуже установленоanacondaокружающая среда, и являетсяgpuверсия, бери!

резюме

Весь процесс компиляции по-прежнему очень гладкий, мы скомпилировали его раньшеopencv,tensorflow,caffe, по сравнению с ними,pytorchОн очень хорошо поработал с зависимостями, был включен в собственный проект в виде подпроектов, во всем процессе компиляции не было ошибок, документация также написана просто и понятно. , Неудивительно, что его доля на рынке будет становиться все выше и выше.

использованная литература

Категории