search

YuvalNirkin / fsgan

FSGAN - Official PyTorch Implementation
https://nirkin.com/fsgan
Creative Commons Zero v1.0 Universal
749 stars 147 forks source link

Assertion Error: Torch not Compiled With CUDA Enabled #151

Open H00plyha opened 2 years ago

H00plyha commented 2 years ago

Hi All,

I am recieving the following error when attempting to run the face_swap_image2video.py script:

image

I have followed the windows install instructions and have CUDA 11.0 installed, alongside torchvision 0.11.1.

Everything is pointing to CUDA correctly in PATH.

Anyone have any thoughts?

honeboy commented 2 years ago

try this

pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html

H00plyha commented 2 years ago

Thank you - will give that a go!