ai:pytorch
差分
このページの2つのバージョン間の差分を表示します。
両方とも前のリビジョン前のリビジョン次のリビジョン | 前のリビジョン | ||
ai:pytorch [2019/12/29 17:00] – oga | ai:pytorch [2021/11/27 20:45] (現在) – [ROCm 4.3 + PyTorch 1.11] oga | ||
---|---|---|---|
行 33: | 行 33: | ||
+ | Model (Module) の定義は class ~Impl を使い、TORCH_MODULE() マクロで定義する。 | ||
+ | <code cpp> | ||
+ | # | ||
+ | class ModelFCImpl : public torch:: | ||
+ | torch:: | ||
+ | torch:: | ||
+ | ~ | ||
+ | public: | ||
+ | ModelFC() | ||
+ | { | ||
+ | fc1= register_module( " | ||
+ | fc2= register_module( " | ||
+ | ~ | ||
+ | } | ||
+ | torch:: | ||
+ | { | ||
+ | ~ | ||
+ | x= torch:: | ||
+ | x= fc2( x ); | ||
+ | return | ||
+ | } | ||
+ | }; | ||
+ | TORCH_MODULE( ModelFC ); | ||
+ | </ | ||
+ | Shape 定義は NCHW 形式で IntArrayRef を使う。< | ||
+ | TORCH_MODULE() で定義しておけば直接 torch:: | ||
+ | ===== C++ API sample ===== | ||
+ | |||
+ | https:// | ||
+ | |||
+ | |||
+ | |||
+ | |||
+ | ====== RADEON (ROCm) で PyTorch を使う方法 ====== | ||
+ | |||
+ | RADEON で PyTorch の < | ||
+ | |||
+ | |||
+ | * [[https:// | ||
+ | |||
+ | |||
+ | ==== ROCm 4.3 + PyTorch 1.11 (2021/ | ||
+ | |||
+ | * Ubuntu 20.04LTS RADEON Vega 56 | ||
+ | |||
+ | < | ||
+ | $ sudo apt update | ||
+ | $ sudo apt dist-upgrade | ||
+ | $ sudo apt install libnuma-dev | ||
+ | $ sudo reboot | ||
+ | $ sudo apt install wget gnupg2 | ||
+ | $ wget -q -O - https:// | ||
+ | $ echo 'deb [arch=amd64] https:// | ||
+ | $ sudo apt update | ||
+ | $ sudo rocminfo | ||
+ | $ sudo apt install rocm-dkms | ||
+ | $ sudo reboot | ||
+ | $ sudo apt install git cmake | ||
+ | $ sudo apt install python3-pip | ||
+ | $ pip3 install setuptools numpy wheel pyyaml | ||
+ | $ git clone --recursive https:// | ||
+ | $ cd pytorch | ||
+ | $ sudo apt install rccl rocm-libs | ||
+ | $ pip3 install typing_extensions | ||
+ | $ python3 tools/ | ||
+ | $ ROCclr_DIR=/ | ||
+ | </ | ||
ai/pytorch.1577606443.txt.gz · 最終更新: 2019/12/29 17:00 by oga