I’m wondering if anyone has built PyTorch 1.2.0 for the Jetson Nano. Following here (https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano/), I was able to install 1.1.0, but have quickly run into issues described here (https://devtalk.nvidia.com/default/topic/1048889/jetson-nano/pytorch-support/2).
The exact error was as follows:
RuntimeError: cuda runtime error (7) : too many resources requested for launch at /media/nvidia/WD_BLUE_2.5_1TB/pytorch-v1.1.0/aten/src/THCUNN/generic/SpatialUpSamplingBilinear.cu:67
It seems that there’s a fix on GitHub (https://github.com/pytorch/pytorch/issues/8103#issuecomment-514802040) that requires modifying and recompiling torch from (1.1.0) source. However, a comment on the second devtalk post from above suggested that simply installing torch from (github/master) source without modifications fixes the issue, and I’m wondering if this is patched in 1.2.0.
If no one has a built wheel for 1.2.0, I’ll do so and check if these issues are fixed in that release; maybe others would be able to use this.