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
更多推荐
所有评论(0)