<p dir="auto">The first time I built it successfully (build_windows.bat), but it didn't seem to find CUDA or cuDNN. I deleted the build directory and tried<br>
set USE_CUDA=ON<br>
It seemed to find the GPU locations successfully. However this error eventually popped up:</p>
<p dir="auto">"C:\Users\pazzu\Documents\pytorch\build\ALL_BUILD.vcxproj" (default target) (1) -&gt;<br>
"C:\Users\pazzu\Documents\pytorch\build\caffe2\caffe2_gpu.vcxproj" (default target) (26) -&gt;<br>
(CustomBuild target) -&gt;<br>
Error : Internal Compiler error (codegen): "there was an error in verifying the lgenfe output!" [C:\Users\pazz<br>
u\Documents\pytorch\build\caffe2\caffe2_gpu.vcxproj]</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="37821 Warning(s)
1 Error(s)"><pre class="notranslate"><code class="notranslate">37821 Warning(s)
1 Error(s)
</code></pre></div>
<p dir="auto">Time Elapsed 01:50:03.60<br>
"Caffe2 building failed"</p>
<p dir="auto">I'm using VS 15 2017 Win64 for the compiler.</p>
<p dir="auto">Can we have <code class="notranslate">unique</code> to return unique rows and cols as well, similar to <code class="notranslate">numpy.unique</code>? Example usage to give an idea:</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="&gt;&gt;&gt; a = torch.tensor([[1, 0, 0], [1, 0, 0], [2, 3, 3]])
&gt;&gt;&gt; torch.unique(a)
tensor([0, 1, 2, 3])
&gt;&gt;&gt; torch.unique(a, dim=0)
tensor([[1, 0, 0], [2, 3, 3]])
&gt;&gt;&gt; torch.unique(a, dim=1)
tensor([[1, 0], [1, 0], [2, 3]])"><pre class="notranslate"><code class="notranslate">&gt;&gt;&gt; a = torch.tensor([[1, 0, 0], [1, 0, 0], [2, 3, 3]])
&gt;&gt;&gt; torch.unique(a)
tensor([0, 1, 2, 3])
&gt;&gt;&gt; torch.unique(a, dim=0)
tensor([[1, 0, 0], [2, 3, 3]])
&gt;&gt;&gt; torch.unique(a, dim=1)
tensor([[1, 0], [1, 0], [2, 3]])
</code></pre></div>