}catch(CuDNNError& e) { TORCH_WARN("Plan failed with a CuDNNError:", e.what()); }catch(c10::OutOfMemoryError& e) { (void)cudaGetLastError();//clear CUDA error TORCH_WARN("Plan failed with an OutOfMemoryError:", e.what()); ...
Hello! Thanks for this fantastic repo, have you checked the python API? I tried on my jetson the following command: python inference.py And I got this error: AssertionError: Torch not compiled with CUDA enabled So I decided to go step by step and when it tries to load de "yolov5.engi...
2检查cuDNN Jetson-nano中已经安装好了cuDNN,并有例子可供运行,我们运行一下例子,也正好验证上面的CUDA cd /usr/src/cudnn_samples_v8/mnistCUDNN/sudo make#编译一下例子 ./mnistCUDNN 如果以上无法运行可以添加权限如下方法: sudo chmod a+x mnistCUDNN# 为可执行文件添加执行权限如果成功,如下所示 更新PIP...
/usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8.9.7 HIP runtime version: N/A MIOpen runtime version: N/A Is XNNPACK available: True CPU: Architecture: x86_64 CPU op-mode(s): 32-bit, 64-bit Address sizes: 48 bits physical, 48 bits virtual Byte Order: Little Endian CPU(s):...
usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8.9.4 HIP runtime version: N/A MIOpen runtime version: N/A Is XNNPACK available: True CPU: Architecture: x86_64 CPU op-mode(s): 32-bit, 64-bit Address sizes: 46 bits physical, 48 bits virtual Byte Order: Little Endian CPU(s): ...
cudnn/lib/libcudnn_cnn_infer.so.8#2 0x00007ffe5e869c9a in cudnn::cnn::Fft_1d_Tiled<float, float, float2, float>::execute_internal_impl(cudnn::backend::VariantPack const&, CUstream_st*) ()from /home/chisari/miniconda3/envs/diffusion_broken_env/lib/python3.10/site-packages/torch/...
TORCH_WARN("Plan failed with a CuDNNError:", e.what()); }catch(c10::OutOfMemoryError& e) { (void)cudaGetLastError();//clear CUDA error TORCH_WARN("Plan failed with an OutOfMemoryError:", e.what()); } } TORCH_CHECK( false, ...