search

jiupinjia / stylized-neural-painting

Official Pytorch implementation of the preprint paper "Stylized Neural Painting", in CVPR 2021.
https://jiupinjia.github.io/neuralpainter/
Creative Commons Zero v1.0 Universal
1.56k stars 262 forks source link

how can i run this program on GPU? #37

Closed boboqazqaz closed 3 years ago

boboqazqaz commented 3 years ago

My GPU is gtx1070. Can I run this program on GPU? When I run this program. The CPU is 100% but GPU only 5%，and also rendering is very slowly Is that right？

jiupinjia commented 3 years ago

Hi @boboqazqaz, thanks for your feedback, it seems you were not using your GPU. Did you install Pytorch-GPU and Cuda correctly?

boboqazqaz commented 3 years ago

IT seems my cudatoolkit isn't right, My GPU driver is 462.30 ，CUDAdrier is 11.2 but i use “conda install” just can install cudatoolkit 10.2 I have to install cudatoolkit 11.2？