ai:pytorch
差分
このページの2つのバージョン間の差分を表示します。
両方とも前のリビジョン前のリビジョン次のリビジョン | 前のリビジョン次のリビジョン両方とも次のリビジョン | ||
ai:pytorch [2020/01/01 22:32] – [メモ] oga | ai:pytorch [2021/11/27 16:30] – [RADEON (ROCm) で PyTorch を使う方法] oga | ||
---|---|---|---|
行 68: | 行 68: | ||
===== C++ API sample ===== | ===== C++ API sample ===== | ||
- | https:// | + | https:// |
+ | ====== RADEON (ROCm) で PyTorch を使う方法 ====== | ||
+ | |||
+ | RADEON で PyTorch の < | ||
+ | |||
+ | |||
+ | * [[https:// | ||
+ | |||
+ | |||
+ | < | ||
+ | $ 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 apt install rocm-dkms | ||
+ | $ sudo reboot | ||
+ | $ sudo apt install cmake | ||
+ | $ sudo apt install python3-pip | ||
+ | $ pip3 install setuptools numpy | ||
+ | $ git clone --recursive https:// | ||
+ | $ cd pytorch | ||
+ | $ sudo apt install rccl rocm-libs | ||
+ | $ pip3 install typing_extensions | ||
+ | $ python3 tools/ | ||
+ | $ ROCclr_DIR=/ | ||
+ | </ | ||
ai/pytorch.txt · 最終更新: 2021/11/27 20:45 by oga