1、几个重要的网址
下载对应的cuda
CUDA Toolkit 11.6 Downloads | NVIDIA Developer
Install Docker Engine on Ubuntu | Docker Documentation
跟着上面网址一步一步操作,
先卸载之前的docker 内容
sudo apt-get remove docker docker-engine docker.io containerd runc
然后照着操作Install Docker Engine这个模块:
sudo apt-get update
sudo apt-get install docker-ce docker-ce-cli containerd.io
然后follow 下面网址更改docker 镜像存储的位置
Install Docker Engine on Ubuntu | Docker Documentation
起docker服务时,使用以下命令:
sudo docker run --name new_docker --gpus all --shm-size 64g -it -v path:path imageid
进入docker,使用如下命令:
sudo docker exec -it new_docker bash