在linux中的docker下搭建cuda环境

显卡驱动安装

根据自己的需要来安装相应的驱动版本

sudo apt install nvidia-drivers-440

安装Docker

sudo apt-get remove docker docker-engine docker.io containerd runc
sudo apt-get update
sudo apt-get install -y apt-transport-https ca-certificates curl gnupg-agent
software-properties-common
curl -fsSL https: