美文网首页
linux 多用户多版本cuda管理

linux 多用户多版本cuda管理

作者: 不为什么的人 | 来源:发表于2020-05-18 23:02 被阅读0次

对某用户指定特定版本 cuda

  1. 查看 /usr/local 下的已有的 cuda 版本

    xxxxxxx@xx: ls /usr/local/cuda*
    /usr/local/cuda-10.1:
    bin    cuda-9.2  EULA.txt  include  libnsight  nsight-compute-2019.4.0  nsight-systems-2019.3.7.5  nvvm share  targets  version.txt cuda-10.1  doc       extras    lib64    libnvvp    nsightee_plugins     nvml         samples    src    tools
    
    /usr/local/cuda-9.0:
    bin  doc  extras  include  jre  lib64  libnsight  libnvvp  nsightee_plugins  nvml  nvvm  pkgconfig  samples  share  src  tools  version.txt
    
    /usr/local/cuda-9.2:
    bin  cuda-9.0  cuda9.0  doc  extras  include  jre  lib64  libnsight  libnvvp  nsightee_plugins  nvml  nvvm  pkgconfig  samples  share  src  tools  version.txt
    
  2. 在该用户的 ~/.bashrc 中添加/修改环境变量,以使用 cuda-10.1 为例:

    export PATH=$PATH:/usr/local/cuda-10.1/bin
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64  
    export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-10.1/lib64  
    
  3. 来源:https://zhuanlan.zhihu.com/p/35834028

多版本 cuda 切换

  1. 更改用户的 ~/.bashrc 中的环境变量,以使用 cuda 代替 cuda-10.1

    export PATH=$PATH:/usr/local/cuda/bin
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64  
    export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64  
    
  2. /usr/local/cuda 软链到某一个版本的 cuda,以 cuda-10.1 为例:

    sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda
    
  3. 以上命令运行之后,用户的 cuda 版本就被设定为了 10.1, 如果要更改版本,则必须先删除旧的软链接,然后重新链接

    # 如果要修改 `/usr/local/cudaa` 的链接,必须先删除,再重新链接
    sudo rm -rf cuda
    sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda
    
  4. 来源

相关文章

网友评论

      本文标题:linux 多用户多版本cuda管理

      本文链接:https://www.haomeiwen.com/subject/ojlpohtx.html