17. Linux下在虚拟环境中安装CUDA和CUDNN
这个专栏记录我学习/科研过程中遇到的一些小问题以及解决方案,一些问题可能比较蠢请见谅。自用,仅供参考。
------------------------------------------------------------------------------------
Linux下在虚拟环境中安装CUDA和CUDNN
目标:13. Linux上CUDA的安装 所提到的CUDA安装方法属于全局安装,会对其他用户造成影响。本文将介绍一种使用conda在虚拟环境中安装的方法以解决这一问题。
分析:
全局安装:
安装方法请参考这篇博客:链接
/usr/local
是系统的全局路径,任何用户在 PATH
或 LD_LIBRARY_PATH
中配置指向 /usr/local/cuda
后都可以访问该版本的 CUDA 和其工具(如 nvcc)。如果多个用户需要不同版本的 CUDA,必须通过手动管理符号链接(/usr/local/cuda)或修改每个用户的环境变量。
由于 /usr/local
是全局路径,在该路径下安装会影响整个服务器的所有用户。如果某个用户修改了 /usr/local/cuda
的符号链接或覆盖了库文件,可能导致其他用户的应用程序出错或无法运行。
此外,用这一方法安装或卸载 CUDA 需要管理员权限(sudo)。
虚拟环境中使用 conda 安装:
安装方法请参考这篇博客:链接
虚拟环境中使用 conda 安装是一种用户级安装。安装的 CUDA 和 CUDNN 仅限于虚拟环境中,隔离性好,不会影响其他用户或系统。Conda 在每个虚拟环境中会下载所需的 CUDA 相关库(如 libcuda.so),这些库仅作用于当前环境,外部程序无法干扰。每个 Conda 虚拟环境可以独立配置所需版本的 CUDA 和 CUDNN,避免了版本冲突。
此外,这一方法不需要系统管理员权限,用户可以完全自主控制。