ai:pytorch

差分

このページの2つのバージョン間の差分を表示します。

この比較画面へのリンク

--- ai:pytorch [2020/01/04 17:59] – [C++ API sample] oga
+++ ai:pytorch [2021/11/27 20:45] (現在) – [ROCm 4.3 + PyTorch 1.11] oga
@@ 行 73: / 行 73: @@
+====== RADEON (ROCm) で PyTorch を使う方法 ======
+RADEON で PyTorch の <nowiki>C++ API</nowiki> を使う。
+  * [[https://wlog.flatlib.jp/archive/1/2020-1-4]]
+==== ROCm 4.3 + PyTorch 1.11 (2021/11/27) ====
+  * Ubuntu 20.04LTS RADEON Vega 56
+<code>
+$ sudo apt update
+$ sudo apt dist-upgrade
+$ sudo apt install libnuma-dev
+$ sudo reboot
+$ sudo apt install wget gnupg2
+$ wget -q -O - https://repo.radeon.com/rocm/rocm.gpg.key | sudo apt-key add -
+$ echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/4.3/ ubuntu main' | sudo tee /etc/apt/sources.list.d/rocm.list
+$ 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://github.com/pytorch/pytorch
+$ cd pytorch
+$ sudo apt install rccl rocm-libs
+$ pip3 install typing_extensions
+$ python3 tools/amd_build/build_amd.py
+$ ROCclr_DIR=/opt/rocm/rocclr RCCL_DIR=/opt/rocm/rccl/lib/cmake/rccl/ PYTORCH_ROCM_ARCH=gfx900 hip_DIR=/opt/rocm/hip/cmake/ USE_NVCC=OFF BUILD_CAFFE2_OPS=0 PATH=/usr/lib/ccache/:$PATH USE_CUDA=OFF python3 setup.py bdist_wheel
+</code>

ai/pytorch.1578128365.txt.gz · 最終更新: 2020/01/04 17:59 by oga