/home/ljx/.local/lib/python3.6/site-packages/torch/nn/_reduction.py:44: UserWarning: size_average and reduce args will be deprecated, please use reduction=‘sum’ instead.
warnings.warn(warning.format(ret))
THCudaCheck FAIL file=/local/jetbot/pytorch/aten/src/THCUNN/generic/SpatialDepthwiseConvolution.cu line=202 error=701 : too many resources requested for launch
Traceback (most recent call last):
File “train_ssd.py”, line 342, in
device=DEVICE, debug_steps=args.debug_steps, epoch=epoch)
File “train_ssd.py”, line 123, in train
loss.backward()
File “/home/ljx/.local/lib/python3.6/site-packages/torch/tensor.py”, line 185, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File “/home/ljx/.local/lib/python3.6/site-packages/torch/autograd/init.py”, line 127, in backward
allow_unreachable=True) # allow_unreachable flag
RuntimeError: cuda runtime error (701) : too many resources requested for launch at /local/jetbot/pytorch/aten/src/THCUNN/generic/SpatialDepthwiseConvolution.cu:202
1.6
Hi,
Which PyTorch package do you install?
If you are not using our prebuilt package, please give it a try.
You can find the installation information in the below topic:
Thanks.
1.6.0
Hi,
Which JetPack version do you use?
If you are using JetPack4.6, would you mind testing PyTorch v1.8.0
with torchvision v0.9.0
?
Thanks.
4.4.1
Hi,
Sorry for the late update.
Please noted that you will also need to use the corresponding jetson_inference branch based on the JetPack version.
Have you checkout with -b L4T-R32.4.4
?
Thanks.
32.4.4.1
wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.9.0-cp36-cp36m-linux_aarch64.whl
Please provide this package and send others. Chinese browsers can’t open and download it
Hi @father, here is a mirror on Google Drive to the PyTorch 1.9 wheel:
https://drive.google.com/file/d/1FHQCDllw3-DqiUg2McKVvS1DbBmxlNaA/view?usp=sharing
Here are the other PyTorch wheels for JetPack >= 4.4
https://drive.google.com/drive/folders/1fPzVb6DNxFd-HSsor8N38gy40h4YZa8v?usp=sharing
您好,这个问题解决了吗?我也遇到了同样的错误,pytorch 1.6 jetpack 4.5
I got the same problem,pytorch 1.6 jetpack 4.5,what can i do to make this work?
1.7就好了
非常感谢,我在按照官网安装pytorch1.7时,安装成功,但是在安装torchvision0.8.1时,卡在了一个位置不动了,也没有报错,在torchvision文件夹执行的这个命令python3 setup.py install --user卡在了:building ‘torchvision.image’ extension
Emitting ninja build file /home/kavin/torchvision/build/temp.linux-aarch64-3.6/build.ninja…
Compiling objects…
Allowing ninja to set a default number of workers… (overridable by setting the environment variable MAX_JOBS=N)
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision torchvision # see below for version of torchvision to download
$ cd torchvision
$ export BUILD_VERSION=0.x.0 # where 0.x.0 is the torchvision version
$ python3 setup.py install --user
$ cd …/ # attempting to load torchvision from build dir will result in import error
$ pip install ‘pillow<7’ # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6