概要
機械学習やデータ分析をする上で手元のマシンだけで分析/開発作業が完結することは少ないですよね。GPUのモリモリ載ったリモートサーバ上でDockerなどを利用して開発しているんじゃないかと思います。
VSCodeを起点に、リモート環境でのPython開発について試行錯誤したのでそのまとめです。
なお筆者はVSCodeを使い始めたばかりの初心者ですのでアドバイス等いただけると幸いです。
環境
作業した環境です。この環境に縛られるものではないと思います。
- localマシン：Macbook Pro
- VSCode: v1.48.2
- リモートOS: Ubuntu 20.04 LTS
- Docker
- local: 2.3.0.4
- remote: 19.03.8
- docker-compose(remote): 1.26.0
実現したいこと
せっかくの統合開発環境なので、localマシン上でVSCodeを立ち上げてlocalのVSCodeだけで作業を完結させたい。
- VSCodeからリモートサーバにSSH接続
- 接続先のサーバでDockerコンテナに接続
- ビジュアルにブレイクポイントの設置などをしながらデバッグ
手順
リモートサーバにSSH接続
Remote Development拡張機能を利用します。このExtensionは、SSH、Container、WSLのそれぞれのリモート開発拡張をひとまとめにしたものらしいです（それぞれインストールしても同じぽい）。
参考：https://code.visualstudio.com/docs/remote/ssh#_getting-started
Extensionの導入
Remote Development：https://marketplace.visualstudio.com/items?itemName=ms-vscode-remote.vscode-remote-extensionpack
リモートサーバの追加
コマンドパレットを開き、Remote-SSH: Connect to Hostを選択すると、接続先の選択リストが出ます。
ここからAdd New SSH Hostを選択するとｓｓｈコマンドの入力を求められるので、ターミナルからssh接続するのと同じコマンドを入れます。
初めて接続する場合は、Add New SSH Hostの項目しか無かったと記憶してます。
鍵認証の場合、sshコマンドを打つ際に-iオプションで鍵を指定します。
これらの設定はデフォルトで~/.ssh/configに書かれます。
ポートフォワーディングの設定や鍵ファイルの設定などを直接書くこともできます。（多分ホストの追加も）
接続
ホストを追加した時と同様、コマンドパレットのRemote-SSH: Connect to Hostからリモートサーバを選択します。または、VSCodeの左側にあるショートカットメニューの「RemoteExplorer」からホストを選択してアクセスしてもOKです。
リモートサーバ上のDockerコンテナを操作
リモートサーバに接続したら、次はDockerコンテナに接続します。
こちらは以下の記事を参考にさせていただきました（参考にというかそのままなぞっただけです。なので是非参照元を確認してください）。
VS CodeからSSHでサーバ上のDockerコンテナを操作する：https://qiita.com/_kani/items/1ce36e36b0bcda403be2
コンテナの立ち上げ
VSCodeのターミナルを立ち上げ、dockerコマンドなどでコンテナを立ち上げます。
RemoteDevelopment Extensionにはdockerコンテナの立ち上げから接続までサポートする機能もあるのですが、これはローカルマシンでのdocker操作のようです。なので、リモートサーバでコンテナを立てる際には通常のdockerコマンドで操作します（僕ができなかっただけ？）。
接続のための準備
ポート設定
ポートフォワーディングの設定をします。（これ仕組みがよくわかってないんですよね。。。）
sshのコンフィグを開きます。コマンドパレットを開いてssh configなどと入れると項目が出てきますので、選択します。
該当のリモートホストの設定にLocalForwardの設定を追記します。
Host 192.168.*******
HostName 192.168.*******
IdentityFile ~/.ssh/id_rsa
User *****
LocalForward 23750 /var/run/docker.sock
リモートホストでdocker拡張機能のパラメータ設定
VSCodeからリモートホストに接続します。なお、リモートホスト上でもRemoteDevelopmentの拡張機能を入れとく必要があります。
Preferrenceから、以下のようにパラメータを設定します。タブでRemoteを選択して、Docker.hostと検索すれば出てくるようです。
tcp://localhost:23750
これで準備完了です。
なお、設定画面はコマンドパレットからpreferences Open User Settingなどと入力すると出てきます。
接続
サイドバーのRemote Explorerから、操作します。
上段のプルダウンメニューでContainersを選択すると、コンテナのリストが現れるので、接続したいコンテナを選択します。
「Attach to Container」でコンテナに接続します（新しくウィンドが立ち上がる）。
デバッガの利用
VSCodは統合開発環境なので、インタラクティブにブレークポイントを設置するなどしてデバッグができます。しかし、特定の環境ではブレークポイントの設定やステップ実行もできずプログラム終了まで走ってしまう現象が確認できています。
参考：VSCodeでPython仮想環境のデバッグ実行時にブレークポイントで止まらない問題
これでは困るのですが、デバッガの設定ファイル'launch.json'にパラメータを設定することでデバッガの機能が有効になるようです。
パラメータの追加
'launch.json'は、VSCodeのサイドメニューからデバッガを開き、上部の歯車マークを押下することでエディタ画面に表示されます。
このjsonファイルにパラメータを追加します。
"stopOnEntry":true
注意
stopOnEntryを入れる順番に注意が必要です。
argsの前に入れたところ、argsパラメータが適用されず、引数がデバッガに渡されないという現象に遭遇してしばらくはまってしまいました。
まとめ
いくつか試行錯誤して現在は上記の手順に落ち着きました。
もっと良い方法ご存知の方教えていただけると助かります。
特にDockerコンテナを操作する仕組みがよくわかってないのが不安です。