<p>我首先找到所有的<code>AT_CHECK</code>并将它们转换为<code>AT_ASSERT</code>,从而解决了这个问题。检查<code>at_assert_fix</code>分支并没有为我解决问题。我正在使用PyTorch 1.6和CUDA 10.1</p>
<p><a href="https://github.com/daniilidis-group/neural_renderer/search?q=AT_CHECK" rel="nofollow noreferrer">https://github.com/daniilidis-group/neural_renderer/search?q=AT_CHECK</a></p>
<pre><code>(base) mona@mona:~/neural_renderer/neural_renderer/cuda$ vi create_texture_image_cuda.cpp
(base) mona@mona:~/neural_renderer/neural_renderer/cuda$ vi load_textures_cuda.cpp
(base) mona@mona:~/neural_renderer/neural_renderer/cuda$ vi rasterize_cuda.cpp
(base) mona@mona:~/neural_renderer$ pip install .
Processing /home/mona/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=6215781 sha256=fbd9e8ef7340fa71b9fcd69da8ee06a1079ebbe5b4c2c3ce92e8124bc8cea7c5
Stored in directory: /tmp/pip-ephem-wheel-cache-tddilaof/wheels/c6/9b/9b/d2cda4f9ac2127278c21ea5c5e23c3354fe0e63365b7af7842
Successfully built neural-renderer-pytorch
Installing collected packages: neural-renderer-pytorch
Successfully installed neural-renderer-pytorch-1.1.3
(base) mona@mona:~/neural_renderer$ python examples/example1.py
Drawing: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 90/90 [00:04<00:00, 19.37it/s]
</code></pre>
<p><a href="https://github.com/daniilidis-group/neural_renderer/issues/94" rel="nofollow noreferrer">https://github.com/daniilidis-group/neural_renderer/issues/94</a></p>