search

ML-GSAI / EGSDE

Official implementation for "EGSDE: Unpaired Image-to-Image Translation via Energy-Guided Stochastic Differential Equations" (NIPS 2022)
195 stars 11 forks source link

环境 #2

Closed zi1zi closed 1 year ago

zi1zi commented 1 year ago

您好，具体各个包的版本是多少呢，cuda，pytorch，其他包之类

gracezhao1997 commented 1 year ago

conda create -n EGSDE python=3.7 conda activate EGSDE conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=10.2 pip install pyyaml pip install scipy