多版本cuda共存

时间:2019-01-17
本文章向大家介绍多版本cuda共存,主要包括多版本cuda共存使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。
  1. 首先在机器中安装一个版本的cuda,参考链接
  2. 然后再安装另一个版本的cuda时,不要配置环境变量(这样可以保证默认使用第一个版本的cuda),然后安装相应版本的cudnn即可。
  3. 安装了两个版本的cuda之后,需要安装anaconda(安装参考链接)来控制python版本以及tensorflow-gpu版本。(其实是因为tensorflow-gpu版本的不同,才要求cuda版本不同的。)
  4. 假设第一次安装的版本如下:tensorfow-gpu1.4 + cuda8.0 + cudnn6.0 + python3.4(即本机的默认环境),但是你现在需要运行tensorflow-gpu1.6 + cuda9.0 + cudnn7.0 + python3.5。那么你需要执行以下步骤
    • 先用anaconda创建一个虚拟环境:先用anaconda创建一个虚拟环境:conda create -n tensor-1.6 python=3.5
    • 开启tensorflow-gpu1.6的虚拟环境:source activate tensor-1.6
    • 新建启动该虚拟环境时需要运行的脚本:
      mkdir -p /home/username/.conda/envs/cuda_test/etc/conda/activate.d
      vim /home/username/.conda/envs/cuda_test/etc/conda/activate.d/activate.sh
      chmod +x /home/username/.conda/envs/cuda_test/etc/conda/activate.d/activate.sh
      在脚本中输入以下内容,使该虚拟环境启动时,自动切换为cuda9.0:
      ORIGINAL_CUDA_HOME=$CUDA_HOME
      ORIGINAL_LD_LIBRARY_PATH=$LD_LIBRARY_PATH
      export CUDA_HOME=/usr/local/cuda-9.0
      export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
      
    • 新建退出该虚拟环境时需要运行的脚本:
      mkdir -p /home/username/.conda/envs/cuda_test/etc/conda/deactivate.d
      vim /home/username/.conda/envs/cuda_test/etc/conda/deactivate.d/deactivate.sh
      chmod +x /home/username/.conda/envs/cuda_test/etc/conda/deactivate.d/deactivate.sh
      在脚本中输入以下内容,使该虚拟环境退出时,自动切换为本机默认环境(即cuda8.0):
      export CUDA_HOME=$ORIGINAL_CUDA_HOME
      export LD_LIBRARY_PATH=$ORIGINAL_LD_LIBRARY_PATH
      unset ORIGINAL_CUDA_HOME
      unset ORIGINAL_LD_LIBRARY_PATH
      

至此,多版本cuda共存的问题就解决了。如果使用不同版本的tensorflow-gpu,就可以自动切换cuda版本了。如果版本多于两个,利用同样的原理进行配置启动与退出时的规则即可。