728x90
requirements에는 cuda 10.0으로 나와있는데
cuda 10.1로 해야 된다.
정상 동작 확인
pytorch repo
github.com/rosinality/stylegan2-pytorch
rosinality/stylegan2-pytorch
Implementation of Analyzing and Improving the Image Quality of StyleGAN (StyleGAN 2) in PyTorch - rosinality/stylegan2-pytorch
github.com
tensorflow 공식 버전보다 훨씬 좋다고 느껴짐...
'Data-science > deep learning' 카테고리의 다른 글
| keras Data generator custom하게 만들기 (0) | 2020.11.02 |
|---|---|
| conda install vs pip install (0) | 2020.10.31 |
| tensorflow 설치시 주의점 (0) | 2020.10.28 |
| gcc version, g++ version downgrade (8 to 7, on linux) (0) | 2020.10.28 |
| tensorflow 설치시 cuda, cudnn 버전 호환성 (0) | 2020.10.28 |