Jetson TX2 在docker容器中import torch 报错的处理方式

1. Jetson TX2 信息：

驱动版本：JetPack 4.6.1

2. docker信息：

docker 镜像：pull 了 nvcr.io/nvidia/l4t-ml:r32.7.1-py3. 该镜像是为JetPack 4.6.1准备的machine learning 镜像

3. 报错：

OSError: libcurand.so.10: cannot open shared object file: No such file or directory

或

OSError: /usr/lib/aarch64-linux-gnu/libcudnn.so.8: file too short

4. 解决办法：

创建容器时，添加命令：

--runtime nvidia

比如：

sudo docker run -dit --name chatglm -p 609:609 -v /home/jetsontx21/chatglm:/workspace -w /workspace --runtime nvidia nvcr.io/nvidia/l4t-ml:r32.7.1-py3