ubuntu20.04使用docker安装caffe-gpu

1.安装nvidia驱动-cuda-cudnn 2.安装docker-nvidiadocker 3.安装caffe 3.1 更改阿里源 1>

sudo chmod -R 777 /etc/docker sudo cp /etc/docker/daemon.json /etc/docker/backup.json sudo gedit /etc/docker/daemon.json

添加如下内容：

{ "registry-mirrors": ["https://alzgoonw.mirror.aliyuncs.com"] }

2>

systemctl restart docker systemctl status docker

出现如下即为成功：

3.2 加载caffe源

sudo docker run -ti bvlc/caffe:gpu caffe --version