nvida/cuda容器编译Caffe踩坑记录(Ubantu16.04+CUDA10.2)
背景
最近再学caffe,同时想自己编译下,了解了解过程。
使用docker容器,利用官方cuda镜像,进行caffe编译,了解编译过程,踩坑记录。
编译使用cmake。
编译caffe-python接口。
官方镜像
#查找cuda镜像
docker search cuda
docker pull nvidia/cuda
注意cuda镜像pull下来默认是最新版的CUDA1...
原创
2019-12-13 10:44:16 ·
1860 阅读 ·
0 评论