ai:pytorch
差分
このページの2つのバージョン間の差分を表示します。
両方とも前のリビジョン前のリビジョン次のリビジョン | 前のリビジョン次のリビジョン両方とも次のリビジョン | ||
ai:pytorch [2021/11/27 16:32] – [RADEON (ROCm) で PyTorch を使う方法] oga | ai:pytorch [2021/11/27 20:26] – [ROCm 4.3 + PyTorch 1.11] oga | ||
---|---|---|---|
行 82: | 行 82: | ||
==== ROCm 4.3 + PyTorch 1.11 ==== | ==== ROCm 4.3 + PyTorch 1.11 ==== | ||
+ | |||
+ | * Ubuntu 20.04LTS RADEON Vega 56 | ||
+ | |||
< | < | ||
$ sudo apt update | $ sudo apt update | ||
行 87: | 行 90: | ||
$ sudo apt install libnuma-dev | $ sudo apt install libnuma-dev | ||
$ sudo reboot | $ sudo reboot | ||
+ | $ sudo apt install git cmake | ||
$ sudo apt install wget gnupg2 | $ sudo apt install wget gnupg2 | ||
$ wget -q -O - https:// | $ wget -q -O - https:// | ||
行 93: | 行 97: | ||
$ sudo apt install rocm-dkms | $ sudo apt install rocm-dkms | ||
$ sudo reboot | $ sudo reboot | ||
- | $ sudo apt install cmake | ||
$ sudo apt install python3-pip | $ sudo apt install python3-pip | ||
- | $ pip3 install setuptools numpy | + | $ pip3 install setuptools numpy wheel pyyaml |
$ git clone --recursive https:// | $ git clone --recursive https:// | ||
$ cd pytorch | $ cd pytorch |
ai/pytorch.txt · 最終更新: 2021/11/27 20:45 by oga