PyTorch+CUDA+Poetry+Pyenvのコンテナ環境をVSCodeから作成する方法 - Qiita

はじめに dockerを使ったcuda+Pytorch+Poetry+Pyenvの環境構築を行いました。 また今回はVSCode上でコーディングしたかったのと、構築した環境の配布も考えていたのでVSCodeの拡張機能であるDevcontainerを使いました。 cuda+Pytorchだけであればnvidia公式からnvidia/cudaのimageを持って来るだけなので簡単なのですが、そこにpoetry, devcontainerの要素が入ってきたため結構複雑なことになりました。 本記事の手法が最適解ではないことは明白ですが、備忘録兼これより良い方法が知りたいのでまとめておきます。 筆者環境 WSL2 on windows cuda 11.6 docker vscode 設定ファイルと解説 # cudaと利用するpytorchに応じて適切なものを選択 FROM nvidia/cuda:1