Linux服务器多cuda如何安装apex

最新推荐文章于 2025-04-27 20:10:49 发布

mantouai

最新推荐文章于 2025-04-27 20:10:49 发布

阅读量1.2k

点赞数 5

CC 4.0 BY-SA版权

分类专栏： apex 文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/mantouai/article/details/120392950

本文介绍了如何在Linux服务器上管理多个CUDA版本，并详细讲解了在CUDA版本切换后安装apex的步骤及解决可能出现的问题，包括检查CUDA版本一致性与手动设置CUDA_HOME。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

设置好linux环境变量

首先实现多个版本cuda的切换

例如我安装的了两个版本的cuda，分别为cuda9.0和cuda10.2。
两个cuda的目录分别为

#cuda9
/data/home/cuiaihao/cuda9
#cuda10.2
/data/home/cuiaihao/cuda10

当我在使用cuda9时，我的环境变量中为

export PATH=/data/home/cuiaihao/cuda9/bin:$PATH
export LD_LIBRARY_PATH=/data/home/cuiaihao/cuda9/lib64:$LD_LIBRARY_PATH

#export PATH=/data/home/cuiaihao/cuda10.2/bin${PATH:+:${PATH}}
#export LD_LIBRARY_PATH=/data/home/cuiaihao/cuda10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

当我在使用cuda10.2时，我的环境变量为

#export PATH=/data/home/cuiaihao/cuda9/bin:$PATH
#export LD_LIBRARY_PATH=/data/home/cuiaihao/cuda9/lib64:$LD_LIBRARY_PATH

export PATH=/data/home/cuiaihao/cuda10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/data/home/cuiaihao/cuda10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

环境