ShopTRAINING/requirements-gpu-cu118.txt

8 lines
206 B
Plaintext

# GPU版本依赖库 (CUDA 11.8版本)
torch==2.7.1 # 将使用PyTorch官方安装命令安装CUDA 11.8版本
numpy==2.3.0
pandas==2.3.0
matplotlib==3.10.3
scikit-learn==1.7.0
tqdm==4.67.1
openpyxl==3.1.5