我已经安装了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)!