目的:训练和测试DIP的性能,熟悉神经网络工作原理,理解底层视觉任务
方法:配置conda环境
步骤:
# 创建conda环境
conda create -n DIP python=3.7
# 激活环境
conda activate DIPSR
#安装依模块
conda install matplotlib
conda install pytorch=1.7 torchvision cudatoolkit=11.3 -c pytorch
conda install tensorboard
conda install -c conda-forge opencv
conda install scikit-image
完成配置即可开始模型训练