search

satoshiiizuka / siggraphasia2019_remastering

Code for the paper "DeepRemaster: Temporal Source-Reference Attention Networks for Comprehensive Video Enhancement". http://iizuka.cs.tsukuba.ac.jp/projects/remastering/
Other
482 stars 102 forks source link

Is there a docker image? #35

Open M4rt1n12 opened 1 year ago

M4rt1n12 commented 1 year ago

Hallo, I'm not able to get the script run on WSL (Win 11) because of the error ModuleNotFoundError: No module named 'torch' . Found some hints on stackoverflow.com but they all don't help. Is there maybe a docker container?

vitacon commented 1 year ago

@M4rt1n12 Is there a reason why are you using WSL? It works in Windows too.

This worked for me in Anaconda:

conda create -n remastering
conda activate remastering
conda install python=3.9
conda install ffmpeg
conda install opencv
conda install scikit-image
conda install tqdm
conda install pytorch-cuda=11.7 -c pytorch -c nvidia -c conda-forge
conda install pytorch -c pytorch -c nvidia -c conda-forge
conda install torchvision -c pytorch -c nvidia -c conda-forge