Closed QIN2DIM closed 1 month ago
针对不同硬件环境选择安装不同版本的 torch 依赖项,可以通过更复杂的 markers 判断运行设备是否存在可用的GPU
https://python-poetry.org/docs/dependency-specification/#using-environment-markers
不过这里直接按照操作系统划分处理了,也即,macOS 统一走默认 PyPI,其余系统安装 gpu 版本
torch = [
{ version = "^2.3.0+cu118", source = "pytorch-gpu-src" },
{ platform = "darwin", version = "^2" }
]
torchaudio = [
{ version = "^2.3.0+cu118", source = "pytorch-gpu-src" },
{ platform = "darwin", version = "^2" }
]
$ poetry show torch
name : torch
version : 2.3.0+cu118
description : Tensors and Dynamic neural networks in Python with strong GPU acceleration
dependencies
- filelock *
- fsspec *
- jinja2 *
- mkl >=2021.1.1,<=2021.4.0
- networkx *
- nvidia-cublas-cu11 11.11.3.6
- nvidia-cuda-cupti-cu11 11.8.87
- nvidia-cuda-nvrtc-cu11 11.8.89
- nvidia-cuda-runtime-cu11 11.8.89
- nvidia-cudnn-cu11 8.7.0.84
- nvidia-cufft-cu11 10.9.0.58
- nvidia-curand-cu11 10.3.0.86
- nvidia-cusolver-cu11 11.4.1.48
- nvidia-cusparse-cu11 11.7.5.86
- nvidia-nccl-cu11 2.20.5
- nvidia-nvtx-cu11 11.8.86
- sympy *
- triton 2.3.0
- typing-extensions >=4.8.0
required by
- encodec *
- torchaudio 2.3.0
- vector-quantize-pytorch >=2.0
- vocos *
README.md 里可以加上相关使用说明
不是老哥你合的太果断了,你可以测一下,我这边只有 linux 和 win 的设备- -
fix: #54
在当前项目目录下创建虚拟环境
.venv
,安装所有依赖项:(Optional) 额外安装开发依赖:
(Optional) 额外安装测试依赖: