dockerを用いてpytorchの機械学習用のdevelのimageを作成したいのですが、cuda12.2の環境にあったpytorch/pytorchのタグがなくて困っています。cudaのバージョンを変えることはできないのですが、どのようにしたら良いでしょうか？
初学者なので検討外れな質問をしているかもしれませんが、そうでしたらそこも含めて教えてくださると幸いです。
参考:pytorch/pytorchのdockerhub
https://hub.docker.com/r/pytorch/pytorch/tags?page=&page_size=&ordering=&name=
Python 3.10 Pytorch 2.1.0 でいいならですが、NVIDIA NGC の pytorch はいかがでしょうか？
nvidia-container-toolkit が必要です。
下記の pytorch が使用できます。
nvcr.io/nvidia/pytorch:23.10-py3 がCUDA 12.2.2
nvcr.io/nvidia/pytorch:23.09-py3 がCUDA 12.2.1です。
@YuyaHori
Questioner
回答ありがとうございます。
重ねての質問になってしまって申し訳ないのですが、その環境を構築したいとき、Dockerfileにどのように記述すれば良いのか良ければ教えていただきたいです。
NVIDIAがイメージを配布しているので、基本的には下記のように実行する感じですね。
$ docker run --gpus all -it --rm nvcr.io/nvidia/pytorch:23.10-py3
上記のOverviewを確認していただければ、大体の使い方はわかると思います。
@geometry dash breeze NVIDIA NGC を 1 年間使用した後、Python 3.10 Pytorch 2.1.0 の方がまだ優れていることがわかりました。
This answer has been deleted for violation of our Terms of Service.
This answer has been deleted for violation of our Terms of Service.
@slope game 特定の CUDA バージョン用の公式 pytorch/pytorch タグがない場合、CUDA 12.2 で PyTorch 用のカスタム Docker イメージを作成するには、どのような手順を踏めばよいでしょうか?
CUDA12.2でPyTorchを独自にビルドして、pipパッケージを作成したものを使用し、カスタムDockerイメージを作成する手順になると思います。
