环境配置。 - 技术栈

环境配置。

weixin_464078072025-12-21 14:33

conda create -n name python=3.10 -y

conda activate name

pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu121

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv -f https://data.pyg.org/whl/torch-2.5.1+cu121.html

pip install torch-geometric==2.7.0 transformers==4.57.3 accelerate==1.12.0 datasets==4.3.0 sktime==0.16.1 safetensors==0.7.0 tokenizers==0.22.1 einops==0.8.1 numpy==1.24.4 pandas==1.5.3 scikit-learn==1.2.2 scipy==1.10.1 matplotlib==3.7.0 jupyterlab notebook ipykernel networkx==3.3 tqdm==4.67.1 pyyaml contourpy dill sympy -i https://pypi.tuna.tsinghua.edu.cn/simple

复制代码

# 尝试使用清华源安装 PyTorch 核心组件
pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://pypi.tuna.tsinghua.edu.cn/simple

# 补充安装 pyg-lib，并安装其他 scatter/sparse 组件
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.5.1+cu121.html

pip install torch-geometric==2.7.0 transformers==4.57.3 accelerate==1.12.0 \
datasets==4.3.0 sktime==0.16.1 safetensors==0.7.0 tokenizers==0.22.1 \
einops==0.8.1 numpy==1.24.4 pandas==1.5.3 scikit-learn==1.2.2 scipy==1.10.1 \
matplotlib==3.7.0 jupyterlab notebook ipykernel networkx==3.3 tqdm==4.67.1 \
pyyaml contourpy dill sympy numba==0.62.1 llvmlite \
-i https://pypi.tuna.tsinghua.edu.cn/simple