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"]    
}

systemctl restart docker
systemctl status docker

出现如下即为成功：

3.2 加载caffe源

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

版权声明：本文来源CSDN，感谢博主原创文章，遵循 CC 4.0 by-sa 版权协议，转载请附上原文出处链接和本声明。
原文链接：https://blog.csdn.net/m0_37975258/article/details/107129976
站方申明：本站部分内容来自社区用户分享，若涉及侵权，请联系站方删除。

发表于 2021-04-11 07:20:08
阅读 ( 1233 )
分类：docker

ubuntu20.04使用docker安装caffe-gpu

你可能感兴趣的文章

精选的优质文章

0 条评论

官方社群

GO教程

推荐文章

猜你喜欢

随便看看