2024年4月14日
WindowsでAIプログラミング
PoetryでPytorch
はじめに
PoetryでPytorchを扱う方法は複数ありまふが、
今回は、TomlファイルにURLを指定する方法でふ。
サンプルコードはこちらにありまふ(/・ω・)/
https://github.com/supplepentan/penta_poetry_pytorch
実装方法
Poetryの設定
こちらを参照くだふぁい。
Poetryで実装
プロジェクトの作成。今回は、『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 |
PytorchでCUDAの確認
『Poetry shell』で仮想環境起動
|
1 |
poetry shell |
pythonを起動して、pytorchでCUDAを確認
|
1 2 3 |
python import torch print(torch.cuda.is_available()) |
これで『True』が出ればOK、
下の画像を参照くだふぁい。
化学系で博士号を取得したが、
あるとき、これからの時代はプログラミング！、と目覚める。
pythonを用いてデータ解析や機械学習に没頭。
最近は、Pytorchで作ったONNXモデルを、Nuxt3にのせたWebサービスの開発、
ChatGPT や Stable Diffusion に没頭中☆('ω')☆
SNSでフォローする
関連記事
