docker-composeでGPU環境(+PyTorch)を構築する - Qiita

はじめに Google Colabratory便利ですよね。 特に何も設定しなくても無料でGPUを利用できます。 しかし、Colabratoryは不自由な部分もありできたらローカルにGPU環境を用意したいものです。 過去にGPU対応に挑戦したことがありましたが、3度ほど挑戦して失敗。 そしてつい先日WSL2のGPU対応によりWSL2でGPUが認識するようになりました。 設定は以下の記事を参考にしてみてください。 また、WSLのDocker環境構築は以下の記事で行えます。 ここまでで、WSL2上でnvidia-smiをすると認識しています。 今回はここからDockerコンテナ上でGPUを認識するように設定していきます。 GPU: GEFORCE GTX 1650 Ubuntu20.04 LTS Windows 問題 まず、WSL2でGPUが認識する状態でいつも通りにコンテナを立ち上げます。