balthazarneveu / blind-deblurring-from-synthetic-data

MVA ENS Paris Saclay - Image restoration project on deblurring learnt on deadleaves
3 stars 0 forks source link

Train NafNet on telecom remote gpu #15

Open balthazarneveu opened 6 months ago

balthazarneveu commented 6 months ago

conda activate rstor pip3 installe -e . conda config --set ssl_verify no conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia pip install wandb scikit-image numba opencv-python tqdm

âš  pip3 install torch torchvision torchaudio does not work with cuda

Try: conda install nvidia::cuda