search

RameenAbdal / StyleFlow

StyleFlow: Attribute-conditioned Exploration of StyleGAN-generated Images using Conditional Continuous Normalizing Flows (ACM TOG 2021)
http://rameenabdal.github.io/StyleFlow
2.42k stars 342 forks source link

Update requirements to support CUDA>=11.0 #146

Open nakhodnov17 opened 1 year ago

nakhodnov17 commented 1 year ago

Following changes were made:

  1. Unnessesary pycache files were removed
  2. Resolves #34 by implementing @simon-rob propositions
  3. Update environment requirements (both for Docker and conda instalations) to support CUDA>=11.0
Karthik-Ragunath commented 1 year ago

Tested a portion of this PR involving commit 0b00386 as part of stage 3 of the run_ffhq_uv_dataset.sh script, and it works as expected !!!

mikelmh025 commented 4 months ago

Just let you know that image used in the docker file should be updated. This is the closest I found from the docker hub. (Basically using cuda:11.2.2 instead of cuda:11.2.0)

https://hub.docker.com/r/nvidia/cuda/tags?page=1&page_size=&ordering=&name=11.2.2-cudnn8-devel-ubuntu

FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu18.04