ai:pytorch

差分

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

この比較画面へのリンク

--- ai:pytorch [2021/11/27 16:30] – [RADEON (ROCm) で PyTorch を使う方法] oga
+++ ai:pytorch [2021/11/27 20:45] (現在) – [ROCm 4.3 + PyTorch 1.11] oga
@@ 行 80: / 行 80: @@
   * [[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>
@@ 行 90: / 行 94: @@
 $ 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 cmake
+$ sudo apt install git cmake
 $ sudo apt install python3-pip
-$ pip3 install setuptools numpy
+$ pip3 install setuptools numpy wheel pyyaml
 $ git clone --recursive https://github.com/pytorch/pytorch
 $ cd pytorch

ai/pytorch.1637998256.txt.gz · 最終更新: 2021/11/27 16:30 by oga