diff --git a/inpainting/lib/correlation_package/make.sh b/inpainting/lib/correlation_package/make.sh index 1ce8e32..d87a446 100755 --- a/inpainting/lib/correlation_package/make.sh +++ b/inpainting/lib/correlation_package/make.sh @@ -1,5 +1,5 @@ #!/usr/bin/env bash -TORCH=$(python -c "import os; import torch; print(os.path.dirname(torch.__file__))") +TORCH=$(python3 -c "import os; import torch; print(os.path.dirname(torch.__file__))") cd src @@ -11,4 +11,4 @@ rm -r ../_ext nvcc -c -o correlation_cuda_kernel.o correlation_cuda_kernel.cu -x cu -Xcompiler -fPIC -arch=sm_52 cd ../ -python build.py +python3 build.py