JetPack 5.1编译mish_cuda

1.查看jetpack版本：sudo jtop

自带的就有cuda11.4和cudnn8.X以及python3.8，我的cudnn就没有是后期自己安装的

2.安装torch

PyTorch for Jetson - Announcements - NVIDIA Developer Forums

选择对应的cuda版本和torch版本，我下载的是：torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64

直接用pip install torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64 安装即可

3.如果系统自带的没有cudnn则需要手动安装：

cuDNN Archive | NVIDIA Developer

选择对应的版本，我选的是 libcudnn8_8.2.2.26-1+cuda11.4_arm64.deb下载

安装：

sudo dpkg -i  libcudnn8_8.2.2.26-1+cuda11.4_arm64.deb

sudo apt-get install libcudnn8=8.2.2.26-1+cuda11.4

sudo apt-get install libcudnn8-dev=8.2.2.26-1+cuda11.4#不用执行

sudo apt-get install libcudnn8-samples=8.2.2.26-1+cuda11.4#不用执行

再次运行sudo jtop即可查看cudnn版本

查看torch

import torch

print("是否可用",torch.cuda.is_available())#显示结果为True即可

在进入下载的mish_cuda目录

运行：sudo python setup.py build install 即可编译成功

本文用到的torch和cudnn已经上传，可以在这里下载：https://download.csdn.net/download/u014793522/88894673