1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
| # 1. 创建环境
conda create -n feddg python=3.9.7
# 2. 激活环境
conda activate feddg
# 3. 安装PyTorch(根据CUDA版本选择)
# CUDA 11.8
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio -c pytorch
# 或CPU版本
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio cpuonly -c pytorch
# 4. 安装其他依赖
conda install numpy=1.20.3
pip install timm
pip install tensorboard
pip install tqdm
pip install einops
# 5. 验证安装
python -c "import torch; print(torch.__version__)"
python -c "import timm; print('timm installed')"
# 6. 导出环境(可选)
conda env export > feddg_environment.yml
|