PyTorch

# 设置虚拟环境在项目目录
poetry config virtualenvs.in-project true
# 创建项目
poetry new pytorch_demo && cd pytorch_demo
# 创建虚拟环境
poetry env use python3
# 进入虚拟环境
poetry shell
# 添加源
poetry source add --priority=primary mirrors https://pypi.tuna.tsinghua.edu.cn/simple/
# 安装基础环境
poetry add torch torchvision torchaudio
poetry add matplotlib
poetry add torch-summary
poetry add pandas

transformers==4.30.2
torch>=2.0