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

出现如下即为成功:
pic0

3.2 加载caffe源

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

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐