方法一
-
宿主机安装NVIDIA驱动和CUDA Toolkit(两者版本要对应)
cuda_10.2.89_440.33.01_linux.run:二合一版本包,包括NVIDIA驱动和CUDA -
容器以特权模式启动:启动时加–privileged,查看容器中/dev目录下是否有nvidia设备
-
在容器内安装NVIDIA驱动(宿主机和容器驱动要一样)
sh **.run --noexec --target nvidia,可以解压二合一版本包得到NVIDIA-Linux-x86_64-440.33.01.run,将驱动拷贝进容器。在容器内执行如下命令安装驱动:
./NVIDIA-Linux-x86_64-440.33.01.run --silent --no-kernel-module --install-compat32-libs --no-nouveau-check --no-nvidia-modprobe --no-rpms --no-backup --no-check-for-alternate-installs --no-libglx-indirect --no-install-libglvnd --x-prefix=/tmp/null --x-module-path=/tmp/null --x-library-path=/tmp/null --x-sysconfig-path=/tmp/null
- 如果要使用CUDA Toolkit,可以将宿主机的CUDA库映射进容器。
或者安装第3步解压得到的cuda-installer(这种方法我没试过)