在Anaconda Prompt完成模型训练

小刘摸鱼中2025-11-14 10:12

单纯是不想配置pycharm的环境

模型：https://github.com/JCruan519/EGE-UNet

1. 创建虚拟环境

从README中获取环境配置信息，创建虚拟环境并配置

bash 复制代码

conda create -n EGE_UNet python=3.8
conda activate EGE_UNet
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html

系统有CUDA 11.8驱动，可向下兼容CUDA 11.1的运行库

Conda会安装CUDA 11.1的运行时工具包，与系统驱动配合使用

PyTorch 1.8.0会使用这个11.1的工具包，而不会与系统CUDA 11.8冲突

如果不行的话，使用

pip install torch==1.8.0 torchvision==0.9.0

安装所需的一些包（话说为什么没有requirements.txt）

bash 复制代码

pip install tensorboardX==2.5.1 timm==0.4.12 opencv-python==4.5.5.64 pillow==9.5.0 matplotlib==3.5.3 scikit-learn==1.0.2 scikit-image==0.19.3 tqdm==4.64.0

2. 数据集

根据README提示将自己的数据放进去

3. 开始训练

cd打开模型文件夹根目录

bash 复制代码

python train.py