本博客是作者复现《S3Net: A Single Stream Structure for Depth Guided Image Relighting》时安装代码环境时的笔记。
1、windows服务器的cuda版本为11.1
安装对应版本的pytorch
1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
2、安装W_SSIM损失函数所需的pywt库
1
conda install pywt
3、安装skimage
1
conda install skicit-image