Anacondaでpytorchをインストールするまで

未来の私のために、備忘録。

1. Anacondaのセットアップ

wgetして、

wget https://repo.anaconda.com/archive/Anaconda3-2023.09-0-Linux-x86_64.sh

ダウンロードしたファイルのハッシュ値を

sha256sum Anaconda3-2023.09-0-Linux-x86_64.sh

以下のサイトの値と比較。

Index of /

実行して、展開。

bash Anaconda3-2023.09-0-Linux-x86_64.sh

baseを最新化しておきましょう。

conda update -n base -c defaults conda

2. 仮想環境の作成

たとえば、python 3.12の環境を作る場合。

conda create -n p312 python=3.12

アクティベート。

conda activate p312

仮想環境のトップディレクトリに移動するには、

cd $CONDA_PREFIX

仮想環境から抜けるときは、

conda deactivate

3. パッケージのインストール

チャネルの追加。

conda config --add channels pytorch
conda config --add channels conda-forge
conda config --add channels nvidia

追加したチャネルの確認。

$ conda config --show channels
channels:
  - nvidia
  - conda-forge
  - pytorch
  - defaults

pytorchのインストール

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia