arm架构docker安装cuda

以下是在arm架构上安装cuda的步骤：
1. 首先，需要安装nvidia-container-runtime，可以使用以下命令进行安装：

sudo apt-get update
sudo apt-get install nvidia-container-runtime

2. 然后，需要下载cuda的安装包，可以在官网上下载对应版本的安装包，例如cuda 10.2：

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

3. 下载完成后，需要给安装包添加执行权限，并运行安装程序：

chmod +x cuda_10.2.89_440.33.01_linux.run
sudo ./cuda_10.2.89_440.33.01_linux.run

4. 在安装过程中，需要根据提示进行选择，例如是否安装驱动程序、是否安装cuda toolkit等。需要注意的是，在arm架构上，只能安装cuda toolkit，不能安装cuda driver。
5. 安装完成后，需要将cuda的库文件路径添加到LD_LIBRARY_PATH环境变量中：

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

6. 最后，可以使用nvidia-smi命令来验证cuda是否安装成功：

nvidia-smi