Skip to content
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

If you are using PyTorch 1.6 #6

Closed
monajalal opened this issue Dec 6, 2020 · 0 comments
Closed

If you are using PyTorch 1.6 #6

monajalal opened this issue Dec 6, 2020 · 0 comments

Comments

@monajalal
Copy link

You would need to change all AT_CHECK in neural mesh renderer to TORCH_CHECK

(base) mona@mona:~/research/phosa/external/multiperson/neural_renderer$ rg AT_CHECK
neural_renderer/cuda/load_textures_cuda.cpp
15:#define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x " must be a CUDA tensor")
16:#define CHECK_CONTIGUOUS(x) AT_CHECK(x.is_contiguous(), #x " must be contiguous")

neural_renderer/cuda/create_texture_image_cuda.cpp
13:#define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x " must be a CUDA tensor")
14:#define CHECK_CONTIGUOUS(x) AT_CHECK(x.is_contiguous(), #x " must be contiguous")

neural_renderer/cuda/rasterize_cuda.cpp
69:#define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x " must be a CUDA tensor")
70:#define CHECK_CONTIGUOUS(x) AT_CHECK(x.is_contiguous(), #x " must be contiguous")
(base) mona@mona:~/research/phosa$ pip install external/multiperson/neural_renderer
Processing ./external/multiperson/neural_renderer
Building wheels for collected packages: neural-renderer-pytorch
  Building wheel for neural-renderer-pytorch (setup.py) ... done
  Created wheel for neural-renderer-pytorch: filename=neural_renderer_pytorch-1.1.3-cp37-cp37m-linux_x86_64.whl size=6321659 sha256=5e2f4afc2346a90c5cd804b226dd7c424ab95f477aad67f2ca3f15530484fbc6
  Stored in directory: /tmp/pip-ephem-wheel-cache-_rf6c5ld/wheels/c7/1b/84/10bf7a286a267887d8c7d382677c292cf18e1bba4e2508ed33
Successfully built neural-renderer-pytorch
Installing collected packages: neural-renderer-pytorch
  Attempting uninstall: neural-renderer-pytorch
    Found existing installation: neural-renderer-pytorch 1.1.3
    Uninstalling neural-renderer-pytorch-1.1.3:
      Successfully uninstalled neural-renderer-pytorch-1.1.3
Successfully installed neural-renderer-pytorch-1.1.3

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant