Torch 世界

借助 PyTorch 探索有趣之事。

支持 PyPI

需要安装 torch-book 包：

pip install torch-book

conda create -n ai python=3.12
conda activate ai
pip install -e .[doc,dev]

安装 PyTorch GPU 版本（其他版本见 PyTorch 安装指南）:

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

pip install .[doc]
inv doc.clean
inv doc