PyTorch

Материал из ALT Linux Wiki

Установка PyTorch

1. Для того чтоб использовать максимально наши пакеты, под root:

 # apt-get update
 # apt-get install cmake \
             python3-module-certifi \
             python3-module-colorama \
             python3-module-filelock \
             python3-module-idna \
             python3-module-jinja2 \
             python3-module-markupsafe \
             python3-module-mpmath \
             python3-module-networkx-core \
             python3-module-numpy \
             python3-module-packaging \
             python3-module-Pillow \
             python3-module-pip \
             python3-module-requests \
             python3-module-setuptools \
             python3-module-sympy \
             python3-module-tqdm \
             python3-module-urllib3 \
             python3-module-wheel

2. Или для того чтоб минимально использовать наши пакеты, под root:

 # apt-get update
 # apt-get install python3-module-pip python3-module-numpy

3. Под вашим пользователем:

 $ pip3 install torch

4. Проверка:

 $ python3 -c 'import torch; print(torch.rand(5,3))'
 tensor([[0.3548, 0.3189, 0.7873],
         [0.6449, 0.2050, 0.7356],
         [0.6996, 0.6416, 0.0680],
         [0.2474, 0.8375, 0.8599],
         [0.1222, 0.2195, 0.0963]])

Проверка доступна ли CUDA:

 $ python3 -c 'import torch; print(torch.cuda.is_available())'
 False