「项目复现」S3net的环境部署记录

Posted on Edited on

​ 本博客是作者复现《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