minicondaでPyTorch環境セットアップ

1. minicondaを入れる

↓を参考にやるだけ

Anacondaとminicondaの違いは、Anacondaはいろいろと高級なツール？がおまけでついてくるらしいのですが、Pythonの仮想環境作るだけならミニマルなminicodnaで十分です。

conda create -n 環境名 python=バージョン

y/n?とか聞かれるので y

Python 3.11で作る場合:

conda create -n py311 python=3.11

(環境名はなんでもいいけど短いとアクティベートするときに楽)

conda 経由で入れます。

↓から好きなバージョンのやつを選びます。

普段使いであればバージョンが高いやつでいいと思います。(基本的に互換性があるため ^[1])

環境のアクティベートを忘れずに

conda activate py311

例

conda install nvidia/label/cuda-12.5.0::cuda-toolkit

nvidia/label/cuda-12.5.0:: はチャンネルを指定しています。

脚注

普段使いであれば安定の新しいやつを入れるといいと思います。

例

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

現在の conda 環境を抜けるには

conda deactivate