search

eladrich / pixel2style2pixel

Official Implementation for "Encoding in Style: a StyleGAN Encoder for Image-to-Image Translation" (CVPR 2021) presenting the pixel2style2pixel (pSp) framework
https://eladrich.github.io/pixel2style2pixel/
MIT License
3.19k stars 570 forks source link

ninja: build stopped: subcommand failed. #4

Closed moonie6568 closed 3 years ago

moonie6568 commented 4 years ago

Thanks for your great work! I have some problems during training the psp. I got the error RuntimeError: Error building extension 'fused': [1/2] :/usr/local/cuda-9.0:/usr/local/cuda-10.0/bin/nvcc -DTORCH_EXTENSION_NAME=fused -D TORCH_API_INCLUDE_EXTENSION_H -isystem /hd1/lvyueming/anaconda3/envs/psp_env/lib/python3.6/site-packages/torch/include -isystem /hd1/lvy ueming/anaconda3/envs/psp_env/lib/python3.6/site-packages/torch/include/torch/csrc/api/include -isystem /hd1/lvyueming/anaconda3/envs/ps p_env/lib/python3.6/site-packages/torch/include/TH -isystem /hd1/lvyueming/anaconda3/envs/psp_env/lib/python3.6/site-packages/torch/incl ude/THC -isystem :/usr/local/cuda-9.0:/usr/local/cuda-10.0/include -isystem /hd1/lvyueming/anaconda3/envs/pspenv/include/python3.6m -D GLIBCXX_USE_CXX11_ABI=0 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constex pr -gencode=arch=compute_75,code=sm_75 --compiler-options '-fPIC' -std=c++11 -c /home/lvyueming/pixel2style2pixel/models/stylegan2/op/fu sed_bias_act_kernel.cu -o fused_bias_act_kernel.cuda.o FAILED: fused_bias_act_kernel.cuda.o :/usr/local/cuda-9.0:/usr/local/cuda-10.0/bin/nvcc -DTORCH_EXTENSION_NAME=fused -DTORCH_API_INCLUDE_EXTENSION_H -isystem /hd1/lvyueming/ anaconda3/envs/psp_env/lib/python3.6/site-packages/torch/include -isystem /hd1/lvyueming/anaconda3/envs/psp_env/lib/python3.6/site-packa ges/torch/include/torch/csrc/api/include -isystem /hd1/lvyueming/anaconda3/envs/psp_env/lib/python3.6/site-packages/torch/include/TH -is ystem /hd1/lvyueming/anaconda3/envs/psp_env/lib/python3.6/site-packages/torch/include/THC -isystem :/usr/local/cuda-9.0:/usr/local/cuda- 10.0/include -isystem /hd1/lvyueming/anaconda3/envs/psp_env/include/python3.6m -D_GLIBCXX_USE_CXX11_ABI=0 -DCUDA_NO_HALF_OPERATORS - DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr -gencode=arch=compute_75,code=sm_75 --compiler-opti ons '-fPIC' -std=c++11 -c /home/lvyueming/pixel2style2pixel/models/stylegan2/op/fused_bias_act_kernel.cu -o fused_bias_act_kernel.cuda.o /bin/sh: :/usr/local/cuda-9.0:/usr/local/cuda-10.0/bin/nvcc: No such file or directory ninja: build stopped: subcommand failed.

could you help me fix it ？Thank you very much!

moonie6568 commented 4 years ago

On more question， which verison of cuda do you choose?

yuval-alaluf commented 4 years ago

Hi @moonie6568!
This seems to be most likely a CUDA problem based on the following issues in rosinality's StyleGAN repo: https://github.com/rosinality/stylegan2-pytorch/issues/123 https://github.com/rosinality/stylegan2-pytorch/issues/63 It appears that you need at least CUDA 10 to get ninja to compile correct. (I personally have CUDA 11 installed)
Let me know if the above links are helpful in answering your questions.

eladrich commented 3 years ago

Closing due to inactivity.