ml-exp-env
機械学習実験環境
実行環境作成
poetryの設定 in ubuntu
RUN curl -sSL https://install.python-poetry.org | POETRY_HOME=/opt/poetry python3 -
ENV PATH="/opt/poetry/bin:$PATH"
RUN poetry config virtualenvs.in-project true
ライブラリのインストール
poetry install
Docker
CUDAによりpytorchのインストール方法が異なるので、適宜公式を参照し、インストールしてください。
docker-compose -f docker-compose-cpu.yml up -d
でコンテナを作成し、VS Codeのms-vscode-remote.remote-containers
から開発環境に入る
gpu周り
もし、docker-compose-gpu.yml
における以下の設定で上手くいかない場合
deploy:
resources:
reservations:
devices:
- driver: nvidia
capabilities: [gpu]
以下に変更する。
runtime: nvidia
vscode extensionの設定
- view/command palletを開き、shellからcodeをインストール
- 新しいshellを開く
- 以下のコマンド実行 (権限は与えておく)
./.devcontainer/vscode_extentions_install_batch.sh