2024年4月14日
PoetryでPytorchを扱う方法は複数ありまふが、
今回は、TomlファイルにURLを指定する方法でふ。
サンプルコードはこちらにありまふ(/・ω・)/
https://github.com/supplepentan/penta_poetry_pytorch
こちらを参照くだふぁい。
プロジェクトの作成。今回は、『penta_poetry_pytorch』という名前で進めまふ。
1 |
poetry new penta_poetry_pytorch |
カレントディレクトリへ移動して、VSCode起動
1 2 |
cd penta_poetry_pytorch code . |
『pyproject.toml』の『[tool.poetry.dependencies]』に、『pytorchのwhl』のURLを追記しまふ。
今回は、『Python 3.8』、『CUDA 11.7』の環境下での追記になりまふ。
1 |
torch = { url = "https://download.pytorch.org/whl/cu117/torch-1.13.0%2Bcu117-cp38-cp38-win_amd64.whl"} |
whlのアドレスは、環境下において変更しくだふぁい。
ちなみに、『%2b』は、『+』を示しまふ。
pyproject.toml
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
[tool.poetry] name = "penta-poetry-pytorch" version = "0.1.0" description = "" authors = ["supplepentan"] readme = "README.md" packages = [{include = "penta_poetry_pytorch"}] [tool.poetry.dependencies] python = "^3.8" torch = { url = "https://download.pytorch.org/whl/cu117/torch-1.13.0%2Bcu117-cp38-cp38-win_amd64.whl"} [build-system] requires = ["poetry-core"] build-backend = "poetry.core.masonry.api" |
1 |
poetry install |
1 |
poetry shell |
1 2 3 |
python import torch print(torch.cuda.is_available()) |
これで『True』が出ればOK、
下の画像を参照くだふぁい。