通过 conda 安装 caffe-gpu 和 caffe
一开始直接使用conda
安装caffe
结果报错。通过不断搜索,最终找到了解决方法:
conda create -n caffe_gpu -c defaults python=3.6 caffe-gpu
或
conda create -n caffe -c defaults python=3.6 caffe
解决方案的原链接在此。安装caffe_gpu
时,对应版本的cudatoolkit
、cudnn
也会被安装。
大佬的原话是:
I will repeat my comment from earlier in this issue, please do not use the anaconda channel when installing caffe, it does not preserve the priority between the main and free sub-channels. Installing caffe or caffe-gpu from the defaults channel is the only method supported by Anaconda and currently works. For example the following two command will create working environments:
conda create -n caffe_gpu -c defaults python=3.6 caffe-gpu
conda create -n caffe -c defaults python=3.6 caffe
意思就是要想通过conda安装caffe
,所有库必须全部使用defaults这个频道,而不能使用或者上混合使用Anaconda 、 conda-forge这些频道的库。
我遇到的问题:
由于一开始 create python
虚拟环境的时候使用conda-forge, 安装caffe
时混合了defaults(第一次)、Anaconda(第二次)的库,结果出现了下面的错误: