用一句命令在Ubuntu 18.04配置anaconda+caffe-gpu+tensorflow-gpu

此前参考了很多篇文博客,他们同时搭建caffe和tensorflow的操作都是用较手动的方式一部部去配置。包括驱动的安装cuda cudnn的安装,这一系列折腾下来还是各种问题,不得其解。
在Ubuntu 18.04的情况下可以使用:

sudo apt install caffe-cuda

的方式去直接安装caffe,但是这种情况下安装的caffe在anaconda环境下是调用不了的。尝试过找到对应的文件复制到anaconda的文件夹里,导入还是会报错,我找不到配置文件也修改不了配置。

一句话安装caffe-gpu + tensorfow-gpu

研究了老半天发觉,anaconda已经是附带caffe的安装了。也就是说安装好anaconda后可以直接conda install caffe-gpu来完成安装,同理可以conda install tensorflow-gpu来安装tensorflow (要注意的是这里都是gpu,而不是caffe-cuda)
这里博主是推荐大家新建一个环境去搭建的
输入

conda create -n env_name -c defaults python=3.7 caffe-gpu tensorflow-gpu 

就可以搭建一个新的名字为env_name的环境了,命令的最后是指定Python版本,我习惯使用的是3.7版本。也可以更改env_name为自己喜欢的环境名。
随后可以通过

conda activate env_name

来激活我们配置好的环境
再输入python进入Python命令行,输入

import caffe
import tensorflow as tf

来试试吧
可能在安装过程中会受网络影响出现失败,建议大家是更换anaconda的源为清华源,参考以下网页,不建议在安装过程中途换源,会引起不必要的错误:
anaconda清华镜像源帮助

猜你喜欢

转载自blog.csdn.net/weixin_43876852/article/details/100743434