欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

多个版本cuda、cudnn切换

程序员文章站 2022-04-18 21:56:14
...
尽量保持最新的CUDA驱动版本
维护多个cuda版本:cuda安装到/usr/local/目录下,可以通过命令切换不同版本



  sudo rm -rf /usr/local/cuda #删除之前创建的软链接
  sudo ln -s /usr/local/cuda-8.0 /usr/local/cuda #创建新 cuda 的软链接

将cuDNN和CUDA解耦:说白了就是,二者单独存放,不要混在一起。切换cuDNN只需要在~/.bashrc文件里面设置如下即可:



 # 在.bashrc文件里面写入该环境变量,切换cudnn只需要把cudnn设置到/root/cuda路径即可
    export LD_LIBRARY_PATH=/root/cuda/lib64:$LD_LIBRARY_PATH