Caffe2无法找到cuda-9.2

时间:2018-11-26 18:32:43

标签: gpu caffe2

我已经安装了cuda-9.0和cuda-9.2。我通常使用cuda-9.0,但是对于caffe2安装,我需要使用cuda-9.2。 我设置了一个包含这些行的文件cuda9.2-env

export PATH=$PATH:/usr/local/cuda-9.2/bin
export CUDADIR=/usr/local/cuda-9.2
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.2/lib64

激活我的conda env后,我会做:

source cuda9.2-env

我的.bashrc文件包含:

export LD_LIBRARY_PATH=/usr/local/cuda/lib64/
export PATH=/usr/local/cuda/

但是caffe2的安装脚本仍然可以检测到cuda-9.0(/ usr / local / cuda)!

0 个答案:

没有答案